1.进入需要安装PyTorch的环境
在开始菜单找到 Anaconda Prompt
点击进入(在cmd
也可)
默认进入是base
环境,首先进入到需要安装 PyTorch
的环境下:
(这里我的环境名为 scl-lle
,修改为自己对应的名称)
conda activate scl-lle
2.切换清华源镜像
复制以下代码执行,切换清华源镜像:
注意:我这里的镜像 url
均为 http
,而非 https
,因为之前试过 https
会报错!
CondaHTTPError: HTTP 000 CONNECTION FAILED for url
如果遇到了类似的错误,可以尝试将https
改为http
并且是 win-64
环境,在末尾加上 win-64
,若是其他系统需修改
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/win-64
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/win-64
conda config --set show_channel_urls yes
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64
注意:若想恢复默认,可执行:
conda config --remove-key channels
此时若直接执行命令安装发现安装还是很慢,还需要进入用户目录下,查看 .condarc
文件
我的目录是:
C:\Users\mi
以记事本打开 .condarc
文件:
若最后有 - default
需要删除并保存,否则在安装时会使用默认 channel
这样镜像就切换成功啦可以进行安装
3.安装对应版本的torch和torchvision
https://pytorch.org/get-started/previous-versions/
找到自己需要安装的torch
和torchvision
版本号:
注意命令最后的 -c pytorch
需要删掉。-c pytorch
的意思是,安装下载的channel
强制为pytorch
官网的channel
。所以需要删除才能走清华镜像的channel
。
即执行:
conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0
等待安装成功即可:
4.验证是否安装成功
最后进行验证是否安装成功:
python
import torch
import torchvision
执行结果如下图所示:
没有报错即安装成功啦,撒花!