Pytorch RuntimeError: couldn‘t find libcudart. Make sure CUDA libraries are installed in a default

# confirm PyTorch sees the GPU
from torch import cuda
assert cuda.is_available()
assert cuda.device_count() > 0
print(cuda.get_device_name(cuda.current_device()))

pytorch gpu无法运行,老是报这个error,重新安装CUDA,pytorch还是无法解决,keras与tensorflow gpu可以使用,最后重新安装Anaconda后成功解决。

>>>
>>> from torch import cuda
>>>
>>> assert cuda.is_available()
>>>
>>> assert cuda.device_count() > 0
>>>
>>> print(cuda.get_device_name(cuda.current_device()))
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "C:\Users\admin\Anaconda3\lib\site-packages\torch\cuda\__init__.py", line 367, in current_device
    _lazy_init()
  File "C:\Users\admin\Anaconda3\lib\site-packages\torch\cuda\__init__.py", line 180, in _lazy_init
    _cudart = _load_cudart()
  File "C:\Users\admin\Anaconda3\lib\site-packages\torch\cuda\__init__.py", line 87, in _load_cudart
    'LD_LIBRARY_PATH'))
RuntimeError: couldn't find libcudart. Make sure CUDA libraries are installed in a default location, or that they're in LD_LIBRARY_PATH.
>>>                          

版权声明:本文为qq_43258953原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。