torch.cuda.is_available() = False的解决方法

1. 首先看下pytorch版本和能否用GPU

    print(torch.__version__)

    print(torch.cuda.is_available())

2. 如果不能,解决方法如下

又名 CUDA和pytorch版本不对应

Step1: 查看CUDA版本
    # 转载https://blog.csdn.net/qq_38295511/article/details/89223169
    # CUDA = 11.0.208 driver

    # 或者cmd打开命令行,输入nvidia-smi

Step2: 查看CUDA和Pytorch的对应关系(可能不准,需要到pytorch官网查看下载torch时对应的CUDA版本)
    # 转载https://www.cnblogs.com/zhoujiayingvana/p/15827369.html



Step3: 如果CUDA版本低,pytorch版本高,就升级电脑的CUDA驱动
    # 官方驱动下载网站: https://www.nvidia.cn/geforce/drivers/
    # 历史版本的驱动程序:
    #     官网:https://www.nvidia.cn/Download/Find.aspx?lang=cn
    #     (法二)转载:https://blog.csdn.net/qq_40493763/article/details/107469539
    # 注意: 虽然GeForce Game Ready看起来是游戏, 但其实是正经的驱动


Step4: 查看CUDA(如11.3)和驱动程序(如512.15)对应关系
    # 官网手动搜索驱动程序:提供的是 驱动程序版本: 512.15 ,不是直接的CUDA版本
    # 官网: https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html


Step4: 发现自动更新后CUDA版本为11.6,pytorch版本为1.11.0
    # 然而目前pytorch最多支持CUDA11.3
    # 现在变成CUDA版本太高了


Step5: 手动下载NVIDIA驱动程序,可以更换电脑的CUDA版本
    # 下载好驱动程序后,直接安装新下载的驱动即可,它自动把老驱动卸载了


其实:超过11.3版本的CUDA,也可以直接pip安装11.3对应的pytorch
高版本CUDA解决方法:(转载)
1. https://www.zhihu.com/question/482335084
2. https://discuss.pytorch.org/t/pytorch-cuda-11-6/149647


Step6: torch和torchvision的对应关系
    # https://github.com/pytorch/vision


Step7: 注意: pip install torch, 下载的是cpu版本的
    # 需要到pytorch官网按照他的命令下载
'''

  • 0
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值