情况
- 安装pytorch版本为pytorch1.8.1+cu11.1,win10,运行时无法调用CUDA,显示初始化失败。
Traceback (most recent call last):
File "anaconda\envs\dp_tf\lib\site-packages\torch\cuda\__init__.py", line 170, in _lazy_init
torch._C._cuda_init()
RuntimeError: CUDA driver initialization failed, you might not have a CUDA gpu.
- 电脑已安装CUDA11.1,但是Pytorch无法调用
- 输入
torch.backends.cudnn.enabled
其输出Out[]: True
- 输入
torch.cuda.is_available()
其输出Out[]: False
解决办法
-
电脑的显卡之前更换为老版本驱动,在更新为最新版本的驱动后,即可正常调用
-
在
Nvidia Geforce Experience
中,更新显卡驱动,即可正常调用。