出现错误(已解决)RuntimeError: CUDA error: no kernel image is available for execution on the device CUDA ker

出现问题(已解决)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://blog.csdn.net/weixin_46065198/article/details/125642546

你可能感兴趣的:(图像分割,深度学习,pytorch,python)