安装win10
安装microsoft visual studio 2015的运行环境(kb4020481),这是安装pytorch的必要条件
安装anaconda3
加入清华源 #由于官方源速度不错,这一步可以不做
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes
安装pytorch
安装依赖包
conda install numpy mkl cffi
conda install pytorch-cpu torchvision-cpu #不能使用CUDA,用CPU做运算
conda install pytorch torchvision #使用CUDA,用GPU做运算
验证PyTorch 安装
在命令行运行python,输入以下代码:
from future import print_function
import torch
x = torch.rand(5, 3)
print(x)

输出结果类似于:
tensor([[0.3380, 0.3845, 0.3217],
                [0.8337, 0.9050, 0.2650],
                [0.2979, 0.7141, 0.9069],
                [0.1449, 0.1132, 0.1375],
                [0.4675, 0.3947, 0.1426]])