RuntimeError:CUDA error:no kernel image is available for execution on the device报错解决(亲测)

今天药忘吃喽~ 2023-09-23 17:53 126阅读 0赞

# 深度学习算法训练报错 #

调试Transformer网络，安装完timm包之后，运行程序时报错`CUDA error:no kernel image is available for execution on the device`，如图所示：

![在这里插入图片描述][1e5b4466e552493bacb315a9357674e7.png_pic_center]

网上对于该错误说啥的都有，因为这是第一次遇到这个错误，之前训练CNN也正常，排除显卡算力低，不支持高版本CUDA问题。看来看去，这位博主说的有道理: [CUDA error: no kernel image is available for execution on the device 报错解决方法][CUDA error_ no kernel image is available for execution on the device]

开始检查自己的pytorch相关包的版本，如图所示：

![在这里插入图片描述][938b3fb5bbf3416fb0cce257649693f2.png_pic_center]

发现问题，我原本torch版本是1.9.1，但是由于安装timm包，torch被重新安装，原版本被覆盖，现在变成1.8.1，所以torch环境被破坏了，包版本之间出现冲突。问题找到了，现在就好办了

# 解决方案 #

按1.8.1版本的torch，以及自己的CUDA版本，重新整理环境。首先去 [pytorch官网][pytorch]对比对应版本的torchvision和torchaudio，以及cudatoolkit。官网的安装命令直接运行，非常慢，建议下载包至本地，然后安装。  
下载链接: [torch相关包][torch]  
安装本地包命令：

pip install 路径+包名

比如，我的CUDA版本是11.1，Python版本是3.7，故下载包：

cu111/torch-1.8.1%2Bcu111-cp37-cp37m-linux_x86_64.whl
    cu111/torchvision-0.9.1%2Bcu111-cp37-cp37m-linux_x86_64.whl

然后安装即可

pip install /路径/cu111/torch-1.8.1%2Bcu111-cp37-cp37m-linux_x86_64.whl

pip install /路径/cu111/torchvision-0.9.1%2Bcu111-cp37-cp37m-linux_x86_64.whl

完美解决报错

**希望本文对大家有帮助，上文若有不妥之处，欢迎指正**

**分享决定高度，学习拉开差距**

[1e5b4466e552493bacb315a9357674e7.png_pic_center]: https://img-blog.csdnimg.cn/1e5b4466e552493bacb315a9357674e7.png#pic_center
[CUDA error_ no kernel image is available for execution on the device]: https://www.cnblogs.com/zhoujiayingvana/p/15655016.html
[938b3fb5bbf3416fb0cce257649693f2.png_pic_center]: https://img-blog.csdnimg.cn/938b3fb5bbf3416fb0cce257649693f2.png#pic_center
[pytorch]: https://pytorch.org/get-started/previous-versions/
[torch]: https://download.pytorch.org/whl/torch_stable.html