The detected CUDA version (XX) mismatches the version that was used to compilePyTorch (XX).

CUDA版本和Pytorch的版本不一致, 对于新手来说真的致命。关于CUDA,CUDA Toolkit ,cuDNN还有pytorch的关系可以参考这篇文章:Pytorch, CUDA,cuDNN的关系 简而言之就是:

nvidia-smi – CUDA Driver API 显卡驱动, 驱动API 也是当前驱动支持的最高CUDA版本,就是平时你看电影打游戏的时候装的驱动,这里是越高越好。
nvcc – CUDA Runtime API CUDA驱动, 用于深度学习做计算的时候用的
CUDA Toolkit – CUDA-C和CUDA-C++编译器和一些科学库
cuDNN – 深度学习的专门CUDA插件库,放在CUDA安装包里就可以了
Pytorch – Python的库,最终的产品

在这里插入图片描述

虽然我的torch.cuda是ture,但是在编译的时候用的主还是CUDA所以这里编译的时候就报错了。其实也很好解决:

  1. 更新torch版本
  2. 更新CUDA版本(我这里选择的是第二步**,因为这里显示我的CUDA版本是10.1而torch是11.6 ,CUDA是向下兼容的所以越高版本的CUDA越兼容低版本的,所以下面我就展示如何把CUDA升到11.6,以WINDOWS11系统为例**)
torch.cuda.is_available()

这里装高级的CUDA,可以参考这一篇文章多CUDA版本切换 , 不用删除之前的CUDA版本,只要下对应的pytorch对应的CUDA版本以及下好对应CUDA版本和cuDNN就可以了,我这里下的就是11.6的CUDA和cuDNN。CUDA默认的路径是,对应的cuDNN下载完,直接放到CUDA安装路径下就可以了。

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6

装完了可以确认下是不是自动配置了环境变量,打开终端输入下面bash指令,就可以看到加入了,那就欧了,如果没有加可以自己配置下新的环境变量就可以了。:

set cuda

在这里插入图片描述

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

曼城周杰伦

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值