pytorch环境配置个人总结

问题

一、关于显卡驱动版

显卡驱动版本过于旧：对应型号参考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版本

解决方法内容来自Pytorch 解决错误ERROR: Could not find a version that satisfies the requirement torch (from versions: none)|极客笔记 (deepinout.com)