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

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