Ubuntu16.0.4下安装深度学习框架：PyTorch

        PyTorch 是Facebook于2017年推出的一个深度学习框架，支持采用 Python 语言编程。PyTorch推出之后大受欢迎，现在已经是 GitHub 上增长速度第二快的开源项目，在过去的 12 个月里，贡献者增加了 2.8 倍，仅次于Tensorflow。2018年12月8日官方已经发布 PyTorch1.0 稳定版。 笔者在此推荐一本书籍叫做《深度学习入门之PyTorch》，索取地址：https://github.com/strivebo/pytorch-learning/tree/master/Books。

下面讲述一下在ubuntu16.0.4上安装PyTorch（支持CUDA的版本）的过程。

必备条件：python、包管理器（anaconda或pip）。笔者推荐anaconda，可以省去很多繁琐的手动安装过程。

一、创建pytorch环境：

conda create -n pytorch

创建完成后激活环境：

conda activate pytorch

二、安装pytorch

打开网站：https://pytorch.org/get-started/locally/，根据自己的情况选择安装命令，pytorch版本建议选择稳定的1.0版本。

执行图中给出的安装命令：

conda install pytorch torchvision cudatoolkit=9.0 -c pytorch

然后会安装很多工具，由于网络原因其中的pytorch—1.0.1可能安装很慢甚至无法安装，建议多试几次，或者尝试配置清华源、阿里源等快速源。然后等待一段时间就安装成功了。

三、安装显卡驱动

参考我的博客：https://blog.youkuaiyun.com/dulingwen/article/details/88565588

四、测试是否安装成功

 1. 为了确保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]])

 2. 另外，为了检查你的GPU和CUDA能够被PyTorch使用，可以执行下面的代码，若返回True，说明CUDA可以使用：

import torch
torch.cuda.is_available()

输出：

True

 

pytorch官方教程：https://pytorch.org/tutorials/index.html