windows conda 安装 PyTorch 切换清华源镜像教程
文章目录
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
执行结果如下图所示:

没有报错即安装成功啦,撒花!
来源:全幼儿园最聪明