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

这篇具有很好参考价值的文章主要介绍了RuntimeError:CUDA error:no kernel image is available for execution on the device报错解决(亲测)。希望对大家有所帮助。如果存在错误或未考虑完全的地方，请大家不吝赐教，您也可以点击"举报违法"按钮提交疑问。

深度学习算法训练报错

调试Transformer网络，安装完timm包之后，运行程序时报错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报错解决(亲测)

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

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

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

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

解决方案

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

完美解决报错

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

分享决定高度，学习拉开差距文章来源地址https://www.toymoban.com/news/detail-513942.html

到了这里，关于RuntimeError:CUDA error:no kernel image is available for execution on the device报错解决(亲测)的文章就介绍完了。如果您还想了解更多内容，请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章，希望大家以后多多支持TOY模板网！