【无标题】

【无标题】 使用清华源安装镜像

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