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