RuntimeError: CUDA error: no kernel image is available for execution on the device CUDA kernel error

最新推荐文章于 2024-06-25 14:36:36 发布

irrationality

最新推荐文章于 2024-06-25 14:36:36 发布

阅读量262

点赞数

分类专栏：机器学习文章标签：深度学习人工智能 python

本文链接：https://blog.csdn.net/weixin_54227557/article/details/131305388

版权

机器学习专栏收录该内容

113 篇文章 3 订阅

订阅专栏

RuntimeError: CUDA error: no kernel image is available for execution on the device
CUDA kernel errors might be asynchronously reported at some other API call,so the stacktrace below might be incorrect.
For debugging consider passing CUDA_LAUNCH_BLOCKING=1.

看起来貌似是匹配的
在这里插入图片描述
但是我的cudnn版本不一定匹配。
安装cudnn
参考：https://blog.csdn.net/h3c4lenovo/article/details/119003405

sudo cp cuda/include/cudnn.h /usr/local/cuda-10.1/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-10.1/lib64
sudo chmod a+r /usr/local/cuda-10.1/include/cudnn.h 
sudo chmod a+r /usr/local/cuda-10.1/lib64/libcudnn*

如果还是不行，继续参考：
https://blog.csdn.net/m0_46483236/article/details/124112298#:~:text=NVIDIA%20A100-PCIE-40GB%20with%20CUDA%20capability%20sm_80%20is%20not,check%20the%20instructions%20at%20Start%20Locally%20%7C%20PyTorch