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为例,首先我们鼠标右键点击“此电脑”,选择“管理”,在计算机管理中,我们点击“设备管理器”,展开“显示适配器”,找到并右键您的显卡设备,选择“更新驱动程序”(也可以选择属性–驱动程序–更新驱动程序),如下图所示。
Windows安装GPU版pytorch报错: UserWarning: CUDA initialization: CUDA unknown error - this may be due to an_第1张图片
2.选择第一个“自动搜索驱动程序”
Windows安装GPU版pytorch报错: UserWarning: CUDA initialization: CUDA unknown error - this may be due to an_第2张图片
3.等待片刻,就可以更新显卡驱动程序,会提示正在联机搜索驱动程序,更新好了之后,会提示“你的设备的最佳驱动程序已安装”。
4.更新好之后,再次验证pytorch,出现True即表示安装成功。

你可能感兴趣的:(python,pytorch,深度学习)