win10安装pytorch

1. 安装anaconda

方便python包的管理

2. 将pip源换成清华源

在C:\Users\lenovo\AppData\Roaming以下文件夹新建一个pip文件，并在文件夹中新建一个pip.ini文件，写入一下内容：

[global]
timeout = 6000
index-url = https://pypi.tuna.tsinghua.edu.cn/simple
trusted-host = pypi.tuna.tsinghua.edu.cn

fig1:

3. 安装pytorch

• 登陆pytorch官网选择相应的环境
  fig2:
  
• copy安装命令到window的cmd中，pip3 install报错则用pip install

pip3 install http://download.pytorch.org/whl/cpu/torch-0.4.1-cp37-cp37m-win_amd64.whl
pip3 install torchvision

4. 测试

• 打开windows的命令行，输入python，再输入以下内容：

import torch    # 测试CPU模式
print(torch.cuda.is_available())      # 输出True则说明GPU可用

fig3:

5. 其他

pytorch在windows下不支持python2.7，如果安装的是anaconda2.7的版本，可以新建一个conda环境，再新的conda环境中安装pytorch

conda create -n pytorch36 python=3.6.1          # 新建名为pytorch36的conda环境，python版本为3.6.1
activate pytorch36          # 激活conda环境
pip3 install http://download.pytorch.org/whl/cpu/torch-0.4.1-cp37-cp37m-win_amd64.whl
pip3 install torchvision    # 安装pytorch