Pytorch_GPU版安装避雷篇

今天被torch折磨，所以写一篇文章记录出过的错，有同感或者正在经历相同痛苦的朋友可以点进来看看

我的配置

显卡是3050

我在NVIDIA控制面板上的CUDA版本可以看到是12.0

我在cmd使用 nvcc -V命令得到的CUDA版本

可以看到是11.1

这两个版本配置不一样的朋友，请你们按照nvcc -V给出的版本进行操作，因为我一开始按照NVIDIA控制面板给的操作失败了。

cudnn版本找不到了，按照网上的教程也没找到，这里就默认是可以和CUDA11.1兼容的版本了（不重要，这个大家应该不会装错）

第一次报错

NVIDIA GeForce RTX 3050 Laptop GPU with CUDA capability sm_86 is not compatible with the current PyTorch installation. The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_61 sm_70 sm_75 compute_37. If you want to use the NVIDIA GeForce RTX 3050 Laptop GPU GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/

错误做法：

在这里我是自动更新了torch的版本，这样做是错误的，遇见这种报错不要更新torch浪费时间，建议直接重装一个，更新之后报了下面的错：

第二次报错

Torch not compiled with CUDA enabled

这个错误表明你当前使用的 PyTorch 版本没有启用 CUDA 支持。要解决这个问题，你需要安装启用了 CUDA 支持的 PyTorch 版本。

 这个时候我们使用如下命令查看torch版本和gpu可用情况

print(torch.cuda.is_available())
print(torch.__version__)#torch版本

如果没有安装正确的torch版本（如安装成CPU版或安装的GPU版与别的什么东西版本不兼容）导致无法使用，就会得到这样的输出：

False
1.8.2+CPU

接下来请卸载掉你之前的torch版本，

pip uninstall torch torchvision torchaudio

然后安装一个新的

这个地方又踩坑了：

我使用了这个代码来安装：

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

这个会根据你当前各种包的版本给你下一个可以兼容的CPU版的torch

很贴心，但没用

正确做法：

当你有了能和你GPU兼容（我的是3050）的CUDA和cudnn后，请你在这个网站上寻找能和它们兼容的torch版本https://pytorch.org/get-started/previous-versions/

我在上面找到了能和我各种硬软件兼容的torch版本

看见我选定的这行字，直接复制然后在后面加上你喜欢的镜像源就能运行。

然后加上中科大源后的完整代码为：

pip3 install torch==1.8.2 torchvision==0.9.2 torchaudio==0.8.2 --extra-index-url https://download.pytorch.org/whl/lts/1.8/cu111 -i https://mirrors.aliyun.com/pypi/simple

按下去，美妙发生

朋友们，一定要注意，你如果要下GPU版的，你的安装包起码要2-3个G，是以G为量级的，如果只有几百几十M就说明下错了！

最后，如果安装过程中出现了上述以外的错误，我建议建个新的虚拟环境重来。因为屎一样的环境是安装包的最大阻力之一。

就是这样，没有结束语，感谢观看。如果觉得有用可以点个赞收藏一下。