使用清华源安装镜像
python3 -m pip install --upgrade x==版本 -i https://pypi.tuna.tsinghua.edu.cn/simple
即可。
安装torchgpu
pip install torch===1.6.0 torchvision===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html
1.安装torch出现以下错误
ERROR: torch has an invalid wheel, .dist-info directory not found
解决方法
pip install -f https://download.pytorch.org/whl/torch_stable.html torch==1.7.0+cpu