跑模型的時候遇到了一個問題
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.
查看了一下當前的cuda版本：
cat /usr/local/cuda/version.txt
CUDA Version 11.0.182
解決方法：
-
將python改為python3.8環境（當時查網上資料有個答主說的，我當時直接改成3.8了，沒有驗證3.7和3.6等是否可行）
-
安裝nightly的pytorch，對應11.0的cuda版本
conda install pytorch torchvision cudatoolkit=11 -c pytorch-nightly
成功解決