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(已经重新安装好了)

torch与pytorch版本不一致 pytorch和cuda版本不匹配_pytorch

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

torch与pytorch版本不一致 pytorch和cuda版本不匹配_深度学习_02

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与pytorch版本不一致 pytorch和cuda版本不匹配_python_03

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

torch与pytorch版本不一致 pytorch和cuda版本不匹配_人工智能_04


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

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

参考 https://www.jianshu.com/p/d1b8bb08b691

然后安装好

最后进行测试是否安装正确

torch与pytorch版本不一致 pytorch和cuda版本不匹配_深度学习_05


表示下载成功,cuda可用。

参考

总结

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