Ubuntu 16.04 pip 安装 Pytorch 0.4.1

之前用Tensorflow比较多，但是也有不得不实现的Pytorch程序，于是动手安装，指定了版本0.4.1，下面记录我的安装过程。

参考：https://blog.csdn.net/gdymind/article/details/82153643

首先需要安装Ananconda3 和 CUDA，我的安装经验在另一篇博客中有所记录，需要的同学可以移步参考。

于是我在Anaconda3和CUDA9.0的环境下开始了Pytorch 的安装。

官网已经推出了Pytorch1.0的最新版本，而且我不知为何无法登录官网的往期版本界面，于是直接用pip命令安装。

网上很多教程用pip3安装，但我在Anaconda3安装完成之后pip命令已经对应python3.6.8的版本，直接使用pip即可。

pip install http://download.pytorch.org/whl/cu90/torch-0.4.1-cp36-cp36m-linux_x86_64.whl

显示信息：

Collecting torch==0.4.1 from http://download.pytorch.org/whl/cu90/torch-0.4.1-cp36-cp36m-linux_x86_64.whl
  Downloading http://download.pytorch.org/whl/cu90/torch-0.4.1-cp36-cp36m-linux_x86_64.whl (519.5MB)
    100% |████████████████████████████████| 519.5MB 7.4MB/s 
Installing collected packages: torch
Successfully installed torch-0.4.1

再查看pip list

多了： torch                              0.4.1    

进入python命令行，输入：

import torch
torch.cuda.is_available()

返回True

继续输入

print(torch.__version__)

返回

0.4.1

说明安装成功。