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/
找到自己需要安装的torchtorchvision版本号:
在这里插入图片描述
注意命令最后的 -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

执行结果如下图所示:
在这里插入图片描述
没有报错即安装成功啦,撒花!

Logo

为开发者提供学习成长、分享交流、生态实践、资源工具等服务,帮助开发者快速成长。

更多推荐