出现问题(已解决)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.
问题——警告
为什么把警告po出来,是因为警告可以让我们了解一些有用信息。
问题——错误
分析原因
首先警告里的内容不可忽略,翻译过来就是NVIDIA RTX GeForce 3060Ti(我使用的服务器)支持的CUDA的算力为8.6,与当前的pytorch的版本不匹配。说白了就是CUDA和pytorch版本不一致。
当前的pytorch版本支持的CUDA的算力为3.7、5.0、6.0、7.0
解决办法
我安装的cuda是11.3,对应的pytorch版本为1.10.1,但是为了跑这个代码,我又安装了pytorch1.10,这就导致了CUDA和pytorch版本不一致
因此我重新安装cuda11.1,然后在pytorch官网找到对应的pytorch1.10的安装语句
pip install torch==1.10.0+cpu torchvision==0.11.0+cpu torchaudio==0.10.0 -f https://download.pytorch.org/whl/torch_stable.html
现在风扇已经在转了!
pytorch官网 链接: https://pytorch.org/get-started/previous-versions/文章来源:https://www.toymoban.com/news/detail-687868.html
参考博客
深度学习环境配置避坑: https://blog.csdn.net/weixin_46065198/article/details/125642546文章来源地址https://www.toymoban.com/news/detail-687868.html
到了这里,关于出现错误(已解决)RuntimeError: CUDA error: no kernel image is available for execution on the device CUDA ker的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!