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

Posted 鲁棒最小二乘支持向量机

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了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,如图所示:

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

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

发现问题,我原本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

完美解决报错

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

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

以上是关于RuntimeError:CUDA error:no kernel image is available for execution on the device报错解决(亲测)的主要内容,如果未能解决你的问题,请参考以下文章

RuntimeError:CUDA 错误:触发了设备端断言 - BART 模型

如何解决“RuntimeError: CUDA error: invalid device ordinal”?

解决RuntimeError: CUDA error: invalid device symbol问题(已解决)

如何解决这个奇怪的错误:“RuntimeError: CUDA error: out of memory”

RuntimeError: cuda runtime error (10) : invalid device ordinal

RuntimeError:CUDA 错误:仅使用 GPU 调用 `cublasSgemm(handle)` 时出现 CUBLAS_STATUS_EXECUTION_FAILED