win10在Anaconda上安装Pytorch

Pytorch官网信息

打开 PyTorch 的官网https://pytorch.org/ , 往下拉找到下如图所示的界面。你需要根据自己机器的情况，回答下面几个问题，以便获得适合你的 PyTorch 安装命令：

• PyTorch Build 版本，你需要安装稳定版还是预览版

• Your OS，你的操作系统是什么

• Package，你准备用什么包管理器

• Language，你准备用什么语言以及什么版本

• CUDA，你的 CUDA的版本是什么

CUDA的安装详细经过和报错处理可以参考：https://blog.csdn.net/wumenglu1018/article/details/88170519

下图就是我当前的环境，即win10+Anaconda+python3.6+cuda9.0

所以安装就直接按照官网提供的命令即可。

创建Pytorch虚拟环境

在安装 PyTorch 之前最好先创建一个 pytorch 的虚拟环境。之所以创建虚拟环境是因为 Python 为不同的项目需求创建不同的虚拟环境非常常见。在实际项目开发中，我们通常会根据自己的需求去下载各种相应的框架库，但是可能每个项目使用的框架库并不一样，或使用框架的版本不一样，这样需要我们根据需求不断的更新或卸载相应的库，管理起来相当麻烦。所以通过创建虚拟环境，相当于为不同的项目创建一块独立的空间，在这个空间里，你安装任何库和框架都是独立的，不会影响到外部环境。

总之就是想象为 PyTorch 单独创建一个空间，这个空间与其它空间互不影响。例如安装TensorFlow的时候也是创建了一个虚拟环境对吧。

打开Anaconda Prompt，输入命令：

conda create --name pytorch python=3.6

这里的pytorch是虚拟环境的名称，可以自己随意取名。

然后进入pytorch虚拟环境：

activate pytorch

安装pytorch

输入上述官网提供的安装命令：

conda install pytorch torchvision cudatoolkit=9.0 -c pytorch

 

安装报错，pytorch包无法下载安装：

CondaHTTPError: HTTP 000 CONNECTION FAILED for url <https://conda.anaconda.org/pytorch/win-64/pytorch-1.0.1-py3.6_cuda90_cudnn7_1.tar.bz2>
Elapsed: -
An HTTP error occurred when trying to retrieve this URL.
HTTP errors are often intermittent, and a simple retry will get you on your way.

因为官方给出的安装命令中最后的-c pytorch，是默认指定anaconda官网上的安装源，可能由于墙的问题，无法正常或者迅速下载。

查看C:\Users\你的用户名下\.condarc文件，如果有-default那一行，就给删掉，并指定清华开源镜像的地址。

我的.condarc文件是没有问题的，所以去掉官网执行命令的-c pytorch，再次执行，发现仍然报错，仍然没有使用指定的清华的安装源。

直接在命令行中指定，将anconda的安装源修改为清华的镜像源：

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/ 
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ conda
config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/ conda 
config --set show_channel_urls yes

然后执行pytorch的安装命令：

conda install pytorch torchvision cudatoolkit=9.0

结果终于成功！！！！！！

验证pytorch的安装

进入python环境：

python

然后，继续输入：

import torch
import torchvision

不报错就表明安装成功

查看pytorch环境：