3090的显卡,需要安装torch1.7.0,cuda11以上版本,否则会报错
pip uninstall 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
3090的显卡,需要安装torch1.7.0,cuda11以上版本,否则会报错
pip uninstall 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