PyToch 安装

PyTorch Get Started
Torch 中文文档

使用 CPU：

conda install pytorch-cpu torchvision-cpu -c pytorch

使用 GPU：CUDA 版本可修改为自己的版本，8.0 / 9.0 / 10.0

# 以cuda9.0版本为例
conda install pytorch torchvision cudatoolkit=9.0 -c pytorch

验证安装成功

python
import torch
import torchvision

如果没有报错就说明安装成功。

安装 PyTorch 包括两部分，一个是 Torch 一个是 torchvision，torchvision 里有很多数据库（比如 MINST）、预训练好的模型，可以直接下载到本地。