Windows安装GPU版pytorch报错: UserWarning: CUDA initialization: CUDA unknown error - this may be due to an
验证GPU版pytorch是否安装成功:
1.打开Anaconda prompt,输入如下代码。
import torch
import torchvision
torch.cuda.is_available()
出现False,报错内容如下:
UserWarning: CUDA initialization: CUDA unknown error - this may be due to an incorrectly set up environment, e.g. changing env variable CUDA_VISIBLE_DEVICES after program start. Setting the available devices to be zero. (Triggered internally at /opt/conda/conda-bld/pytorch_1614378083779/work/c10/cuda/CUDAFunctions.cpp:109.)
return torch._C._cuda_getDeviceCount() > 0
原因:
nvidia-modprobe 版本过低或未安裝,与显卡驱动的版本不匹配。
解决方法:
1.以Win10为例,首先我们鼠标右键点击“此电脑”,选择“管理”,在计算机管理中,我们点击“设备管理器”,展开“显示适配器”,找到并右键您的显卡设备,选择“更新驱动程序”(也可以选择属性–驱动程序–更新驱动程序),如下图所示。
2.选择第一个“自动搜索驱动程序”
3.等待片刻,就可以更新显卡驱动程序,会提示正在联机搜索驱动程序,更新好了之后,会提示“你的设备的最佳驱动程序已安装”。
4.更新好之后,再次验证pytorch,出现True即表示安装成功。