显卡RTX 3090运行pytorch报错CUDA error: no kernel image is available for execution on the device

JY HUA

于 2021-01-12 02:07:58 发布

阅读量9.9k

点赞数 13

CC 4.0 BY-SA版权

文章标签： pytorch

本文链接：https://blog.csdn.net/CallMeYunzi/article/details/112505151

博客提到遇到RuntimeError: CUDA error: no kernel image is available for execution on the device报错，解决办法是针对GeForce RTX 3090显卡安装pytorch，需卸载当前版本的pytorch后重新安装。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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

安装适用于GeForce RTX 3090显卡的pytorch

卸载当前版本的pytorch, 重新按照以下安装

pip uninstall torch

# torch 1.7.0+cu110
pip install torch==1.7.0+cu110 torchvision==0.8.1+cu110 torchaudio===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html

# 更新：现在可使用torch 1.8.2 啦！！
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch-lts -c nvidia