跑模型的时候遇到了一个问题
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
成功解决