torch.cuda.is_available为False的一种可能

问题原因

在Nvidia更新了显卡驱动后,我的pytorch使用不了cuda了,真的非常离谱🤮。吐血收集了一些信息后,这里写了我的解决办法。

nvcc --version 和 nvidia-smi显示不一样

这里引用下一位博主的解释:nvcc和nvidia-smi显示的版本不一致?

CUDA有 runtime api 和 driver api,两者都有对应的CUDA版本, nvcc --version 显示的就是前者对应的CUDA版本,而 nvidia-smi显示的是后者对应的CUDA版本。

通常,driver api的版本能向下兼容runtime api的版本,即 nvidia-smi 显示的版本大于nvcc --version 的版本通常不会出现大问题。

而我更新显卡驱动后,nvidia-smi 显示的driver api为cuda12.0,此时nvcc --version 显示的runtime api为cuda11.7,这样不会有问题,driver api大于runtime api。
nvidia-smi
nvcc --version

conda 和 pip的区别

pytorch官网下面有conda和pip的安装方式,在我无数次使用conda安装卸载pytorch后,决定使用pip来卸载安装,神奇的是问题解决了。
具体原因可能是跟conda和pip的下载位置不同的关系,不清楚具体原理,但在使用conda卸载安装不管用时可以试试pip
pytorch

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值