【ERROR】RuntimeError: CUDA error: no kernel image is available for execution on the device出现问题及解决方案

RuntimeError: CUDA error: no kernel image is available for execution on the device

可能原因一:GPU与CUDA以及Pytorch版本对应有误导致

一般为出现原因为GPU与CUDA以及Pytorch版本对应有误,先采用以下语句测试

torch.cuda.is_available()
a=torch.Tensor([1,2])
a=a.cuda()
a

若以上语句出现问题则需要查看,pytorch与cuda对应版本和GPU是否匹配
pytorch_历史版本
若以上语句测试均无问题则要考虑是否为其他框架导致,如笔者遇到的是MMCV版本与CUDA以及Pytorch版本的对应关系出现了问题

解决方案

重装环境

  1. 先装CUDA 和 cuDNN
  2. 根据CUDA版本安装Pytroch
  3. 根据CUDA和pytorch版本安装MMCV等

可能原因二:MMCV的bug导致

详见:【BUG】MMCV的坑

Original: https://blog.csdn.net/weixin_44493291/article/details/124845742
Author: Chaossll
Title: 【ERROR】RuntimeError: CUDA error: no kernel image is available for execution on the device出现问题及解决方案

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/629628/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球