进入官网,按照你的环境选择,得到下载命令,这里以windows为例,Ubuntu同理

conda安装

官网得到命令如图:

但由于墙的原因直接使用此命令下载很慢,因此请更换conda源为国内镜像
并将命令中 -c pytorch去掉

1
conda install pytorch torchvision cudatoolkit=10.1

即可调用国内镜像下载。

pip安装

官网得到命令如图:

红色框出部分为官网建议版本,但由于墙的原因直接使用此命令下载很慢,因此将pip下载地址改成国内镜像即可

1
pip3 install torch==1.4.0 torchvision==0.5.0 -i https://pypi.tuna.tsinghua.edu.cn/simple