CUDA版本不对重装+离线快速下载torch的gpu版本安装+各种坑
windows走过配环境的坑,记录以下,防止下次再踩
文章目录
- CUDA版本不对重装+离线快速下载torch的gpu版本安装+各种坑
- 一、CUDA版本不对重装
- 1.查看自己电脑目前已经安装的cuda版本
- 2.输入nvidia-smi,是查看自己电脑支持cuda的最高版本,向下兼容
- 3.卸载cuda
- 二、离线快速下载torch的gpu版本安装
- 1.下载torch包
- 2.安装下载好的torch whl文件包
- 总结
一、CUDA版本不对重装
1.查看自己电脑目前已经安装的cuda版本
输入 nvcc -V
我版本是cuda11.1(已经重新安装好了)
2.输入nvidia-smi,是查看自己电脑支持cuda的最高版本,向下兼容
3.卸载cuda
参考这个很详细
卸载完之后,下载cuda
下载cuda时要查看好自己要下载的torch版本的对应的cuda版本是多少,然后再去下载cuda版本。
上pytorch官网查看,https://pytorch.org/get-started/previous-versions/
然后下载完,之后就是按照 进行安装即可
``
二、离线快速下载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可用。
参考
总结
环境有时恶心,有时挂了vpn进行pip下载会报错,踩了坑记录以下。
熟悉了流程就清楚一些bug是出在哪个流程了,知道该怎么解决。