Pytorch:报错RuntimeError: CUDA error: no kernel image is available for execution on the device详细处理报告

关于这个报错的原因,一众文章已经讲得很清楚了,就是pytorch和cuda的版本不匹配。具体的解决方案,直接搜这个报错信息就可以搜出来很多文章,照做即可。本文在此也简单总结一下安装步骤:

  1. 确定自己需要的pytorch版本,然后确定对应的cuda和python版本
  2. 安装正确版本的cuda
  3. 创建正确版本的python环境
  4. 安装正确版本的cudatoolkit
  5. 安装正确版本的pytorch

细节大体上可以参考这篇这篇.

本文主要分享一些干货,即个人踩的一些坑、造成的原因以及对应的解决方法:

  1. 安装pytorch的时候一定要装形如x.y.z+cu111这样的版本,而且要先装好cuda和cudatoolkit再装pytorch,不然会找不到对应库文件,而且报错很隐蔽:使用torch.cuda.is_available()显示true,但是在实际训练的时候会报错上述RuntimeError,导致无法训练。
  2. 我本人是在使用pytorch lightning的时候遇到这个问题的,RuntimeError是在trainer里面触发的,我没办法改pl库里面的代码,只能回过头来解决版本问题;而且同时需要保持所使用的pytorch lightning和torchvision的版本。这种情况方才体现出依赖精确版本号的重要性。
  • 6
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值