GeForce RTX 3090 with CUDA capability sm_86 is not compatible with the current PyTorch installation. The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_70. If you want to use the GeForce RTX 3090 GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/
基本没问题,但是在第二步pip uninstall torch时,需要把torchvision和torchaudio都卸载
pip uninstall torch torchvision torchaudio
然后再
pip install torch==1.10.0 torchvision torchaudio -f https://download.pytorch.org/whl/cu111/torch_stable.html