前提
已经安装好了anacoda
一,打开官网选择合适的torch版本
torch网址
由于我的电脑只有核显所以选择了CPU版本
复制RUN里面生成的pip命令在cmd里面打开就行了,但会遇到下载速度很慢的问题所以这时我们可以修改后面的服务器地址,
只取-f 前的这一段
pip3 install torch==1.8.1+cpu torchvision==0.9.1+cpu torchaudio===0.8.1
在后面加上豆瓣镜像尾椎
-f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.doubanio.com/simple
组合起来就是
pip3 install torch==1.8.1+cpu torchvision==0.9.1+cpu torchaudio===0.8.1 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.doubanio.com/simple