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版本的对应关系出现了问题
解决方案
重装环境
- 先装CUDA 和 cuDNN
- 根据CUDA版本安装Pytroch
- 根据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/
转载文章受原作者版权保护。转载请注明原作者出处!