Ubuntu18.04安装Pytorch

记录自己ubuntu18.04安装pytorch的心得，anaconda3-5.2已经安装好，这里不再详述。

目录

1.CPU版本Pytorch安装

2.GPU版本Pytorch安装

3.相关网址及总结

---

1.CPU版本Pytorch安装

CPU版本的pytoch安装比较简单，在Pytorch官网中选择python版本，运行conda或者pip命令即可，如下图所示。注意，为了方便多个Python版本的管理，通常需要在anaconda中创建新的环境空间。

2.GPU版本Pytorch安装

GPU版本的pytorch安装比较复杂，在安装pytorch之前，通常需要安装显卡驱动，cuda和cudnn，CUDA是NVIDIA推出的用于自家GPU的并行计算框架，也就是说CUDA只能在NVIDIA的GPU上运行；cuDNN（CUDA Deep Neural Network library）：是NVIDIA打造的针对深度神经网络的加速库，是一个用于深层神经网络的GPU加速库。如果你要用GPU训练模型，cuDNN不是必须的，但是一般会采用这个加速库。CUDA是必须的，cudnn是可选的。

通常需要考虑：

（1）检查显卡型号是否支持GPU运算

检查方法：终端输入

ubuntu-drivers devices

得到以下结果：

图中，“model”对应的就是电脑显卡型号，“driver”中后面带有“recommanded”就是推荐的显卡驱动型号。根据显卡型号到英伟达官网查询是否支持GPU。

网站显示如下：

先选择“CUDA GeForce and TITAN Products”，再看具体型号，电脑GPU为GTX 1050，支持GPU 运算。

（2）安装显卡驱动

确定显卡支持GPU 运算后，接下来需要安装显卡驱动，目前尝试过两种安装方法，都成功了。

安装方法一：

系统“软件和更新”，“附加驱动”，选择推荐的驱动安装。

安装方法二：

运行以下命令，安装所有驱动

sudo ubuntu-drive