问题
一、关于显卡驱动版
显卡驱动版本过于旧:对应型号参考CUDA 12.5 Update 1 Release Notes (nvidia.com)
在cmd面板输入NVIDIA-smi
可以查看自己的驱动版本跟cuda的版本,确保N卡驱动版本大于cuda所需要的版本
二、安装的环境中pytorch下载版本不匹配
在anaconda中运行pytorch环境,输入pip list查看自己下载的torch版本
!
在这里下载正确之后我的torch版本是1.10.2,此时运行python环境是可以确定返回true通过的
这是我最开始装错版本时的截图
可以看到版本为1.8.1+cpu,就会导致GPU无法通过
解决方法
一 、安装界面直接输入
pip3 install torch==1.8.1+cpu torchvision==0.9.1+cpu torchaudio===0.8.1 -f https://download.pytorch.org/whl/lts/1.8/torch_lts.html
因为我们电脑都是Windows,如果是Python3.7环境就可以直接下载好我上面那个对我来说是错误的环境
二 、推荐方法
因为我们使用了anaconda安装PyTorch,就可以使用以下命令行
conda install pytorch torchvision torchaudio -c pytorch
就可以直接自动安装适合我们系统的PyTorch版本