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:
win10安装pytorch

3. 安装pytorch

  • 登陆pytorch官网选择相应的环境
    fig2:
    win10安装pytorch
  • 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:
win10安装pytorch

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

相关文章: