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

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

    参考这个很详细 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是出在哪个流程了,知道该怎么解决。

    来源:Bellwen

    物联沃分享整理
    物联沃-IOTWORD物联网 » CUDA版本不对重装+离线快速下载torch的gpu版本安装+各种坑

    发表评论