Ubuntu16.04下安装pytorch全过程

Pytorch的官网地址

目前最新版本的Pytorch版本是1.4，支持最新的cuda9.2及10.1
根据自身的CUDA版本，可以选择适合的Pytorch版本
（千万要选择与自身电脑或服务器CUDA版本所适应的，否则会出错！）
点击下方图片中的红字 install previous versions of PyTorch
即这个网址： https://pytorch.org/get-started/previous-versions/

看到如下界面：

根据需要选择相应的版本命令。
这里我是基于CUDA9.0和Anaconda安装的，故而选择的命令是：

conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=9.0 

这里注意把”-c pytorch”删除，会使用预先配置的国内镜像源，下载速度会快很多，具体配置方法很简单，不过多赘述。

安装完成后，输入：

python

进入anaconda的python环境

然后输入：

import torch
import torchvision
print(torch.__version__)
print(torchvision.__version__)

得到如下显示

可以再输入：

print(torch.cuda.is_available())

显示:True

安装完成！