CUDA版本不对重装+离线快速下载torch的gpu版本安装+各种坑

CUDA版本不对重装+离线快速下载torch的gpu版本安装+各种坑

windows走过配环境的坑，记录以下，防止下次再踩

---

---

一、CUDA版本不对重装

1.查看自己电脑目前已经安装的cuda版本

输入 nvcc -V
我版本是cuda11.1（已经重新安装好了）

2.输入nvidia-smi，是查看自己电脑支持cuda的最高版本，向下兼容

3.卸载cuda

参考这个很详细 https://blog.csdn.net/Zinnir/article/details/122766367
卸载完之后，下载cuda
下载cuda时要查看好自己要下载的torch版本的对应的cuda版本是多少，然后再去下载cuda版本。
上pytorch官网查看，https://pytorch.org/get-started/previous-versions/
然后下载完，之后就是按照https://blog.csdn.net/Zinnir/article/details/122766367 进行安装即可

``

二、离线快速下载torch的gpu版本安装

torch下载直接用官网的pip下载十分慢，会出bug
采用换国内源的方法也会十分慢，也会出bug
所以，直接采用离线下载torch的包，然后再直接安装会快很多

1.下载torch包

上官网，https://pytorch.org/get-started/previous-versions/
查看对应cuda对应的torch版本，然后下载

下载对应版本torch，https://download.pytorch.org/whl/torch_stable.html

此处cu111，代表cuda11.1版本，cp38，代表python3.8.x版本（我是python3.8.5），一定要下对应的版本，python版本跟cuda‘版本没对应上都会在最后安装的时候报错

2.安装下载好的torch whl文件包

参考 https://www.jianshu.com/p/d1b8bb08b691
然后安装好
最后进行测试是否安装正确

表示下载成功，cuda可用。
参考https://blog.csdn.net/zcw1234515/article/details/115050877

总结

环境有时恶心，有时挂了vpn进行pip下载会报错，踩了坑记录以下。
熟悉了流程就清楚一些bug是出在哪个流程了，知道该怎么解决。