报错时安装的CUDA 11.1,Pytorch 1.8.1+cu102
尝试措施1:安装Pytorch 1.10.0
具体操作:
(1)查看机器python版本:
python -V
(2)按照python版本寻找对应的pytorch 和torchvision下载:
download.pytorch.org/whl/torch_stable.htmlhttp://download.pytorch.org/whl/torch_stable.html (3)安装:
pip install torch-1.10.0+cu111-cp39-cp39-linux_x86_64.whl
pip install torchvision-0.11.0+cu111-cp39-cp39-linux_x86_64.whl
成功解决!!!