GeForce RTX 3090 with CUDA capability sm_86 is not compatible with the current PyTorch installation.

问题描述

今天遇到了如题所示的一个问题，原因可能是因为 torch 版本的问题。原先我是直接 pip install torch 默认下载最新版本的 torch。

解决方案

1. 删除原来的 torch

pip uninstall torch

2. 重新安装 torch

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