RuntimeError: CUDA error: no kernel image is available for execution on the device
问题
torch的issue里面确实提到是算力的问题,但是我tm在尊贵的3090ti和A4000上面都遇到了一样的问题,还说是gpu算力跟不上就有点牵强了。
CUDA error: no kernel image is available for execution on the device
问题原链接:github
解决2
首先,清除所有安装的割炬并从正确的来源重新安装。
因为我安装了 CUDA11.1 (和相应的 CUDNN cudnn-11.1-linux-x64-v8.0.5.39 ),所以我使用了 pip3 install --pre torch torchvision -f https://download.pytorch.org/whl/nightly/cu111/torch_nightly.html -U
理想情况下,不应该使用夜间版本,除非您正在寻找当前正在开发的特定内容,特别是当最新版本的 CUDA 尚未被 Torch 支持时。
pip3 install --pre torch torchvision -f https://download.pytorch.org/whl/nightly/<cuda_ver>/torch_nightly.html -U
如果使用的cuda版本为11.3,修改其中的cu111为cu113即可。
当然,这种方法可能也没有用,也有用户在使用这种方法重装以后没有用,那种情况我也不清楚如何解决了,去issue里面找找吧
|