I encountered a problem when running the model

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 .

Check out the current cuda version:

cat /usr/local/cuda/version.txt
CUDA Version 11 . 0 . 182

Solution:

1. Change python to python3.8 environment (At the time, an answer to the online information said, I directly changed it to 3.8 at the time, and did not verify whether 3.7 and 3.6 are feasible)
2. Install nightly’s pytorch, corresponding to the cuda version of 11.0conda install pytorch torchvision cudatoolkit=11 -c pytorch-nightly

Successfully resolved