Ubuntu16.04环境下PyTorch简易安装教程

安装NVIDIA GPU显卡驱动

如果需要安装cuda版本的PyTorch，电脑也有独立显卡的时候，一般需要更新一下Ubuntu独立显卡驱动。否则即使安装了cuda版本的PyTorch也没办法使用GPU。

因为我的笔记本电脑显卡是NVIDIA的，所以首先去官网：NVIDIA显卡驱动下载，查看适合自己显卡的驱动，下载runfile文件：NVIDIA-Linux-x86_64-384.98.run。

然后按Ctrl + Alt + F1到控制台，关闭当前图形环境：

sudo service lightdm stop

卸载可能存在的旧版本NVIDIA 驱动（这一步是必须的，不然重启会有问题）:

sudo apt-get remove --purge nvidia*

安装驱动可能需要的依赖：

sudo apt-get update
sudo apt-get install dkms build-essential linux-headers-generic

把 nouveau 驱动加入黑名单并禁用用 nouveau 内核模块（如果新重装的系统，没有驱动起作用，忽略）：

sudo nano /etc/modprobe.d/blacklist-nouveau.conf

在文件 blacklist-nouveau.conf 中加入如下内容：

blacklist nouveau
options nouveau modeset=0

保存退出，执行：

sudo update-initramfs -u

然后重启：

reboot

重启后再次进入字符终端界面（Ctrl + Alt + F1），并关闭图形界面：

sudo service lightdm stop

进入之前NVIDIA驱动文件下载目录，安装驱动：

sudo chmod u+x NVIDIA-Linux-x86_64-384.98.run
sudo ./NVIDIA-Linux-x86_64-384.98.run -no-opengl-files

-no-opengl-files表示只安装驱动文件，不安装OpenGL文件。这个参数不可忽略，否则会导致登录界面死循环。

最后重新启动图形环境：

sudo service lightdm start

可以通过以下命令确认驱动是否正确安装：

cat /proc/driver/nvidia/version

值得注意的是，安装完显卡驱动后，桌面可能会出现菜单任务栏及窗体最大化最小化消失问题。解决办法如下：

首先安装compiz配置管理器：

sudo apt-get install compizconfig-settings-manager

然后，打开配置管理器：

sudo ccsm

并在”general”启用OpenGL，在”Desktop”里启用Ubuntu Unity Plugin，然后返回桌面。

接着，重置compiz：

dconf reset -f /org/compiz/

最后注销重启：

gnome-session-quit

至此，NVIDIA显卡驱动安装成功。

科学上网的简易方法

进入PyTorch官网：
PyTorch

显示如下：

许多教程说在“Run this command”下会显示安装PyTorch的命令。如果Ubuntu下已经配置过Anaconda的话，可以直接使用conda命令安装成功PyTorch：

conda install pytorch torchvision cuda80 -c soumith

其中，pytorch是软件，torchvision是pytorch使用的数据包和库，cuda80是GPU版本。

但是基于特殊的网络国情，一般无法查看“Run this command”之后的指令，安装也不会成功。

下载PyTorch

不能科学上网的话，使用另外一种方法来安装PyTorch。首先下载所需版本的PyTorch。在PyTorch主页上点击“Click here for previous versions of PyTorch”，如上图右下角所示。

找到所需版本，如上图所示，下载torch-0.2.0.post3-cp36-cp36m-manylinux1_x86_64.whl。其中，0.2.0表示PyTorch版本，cp36表示支持Python3.6，cuda80表示支持GPU版本，电脑有GPU的话须选择该项。

安装PyTorch

进入PyTorch的下载目录，使用pip命令安装：

pip install torch-0.2.0.post3-cp36-cp36m-manylinux1_x86_64.whl

安装torchvision

安装torchvision比较简单，可直接使用pip命令安装：

pip install torchvision

至此，pytorch和torchvision已安装完毕。

更新numpy

安装成功pytorch和torchvision后，打开ipython，输入：

import torch

可能会出现报错的情况，如下所示：

ImportError: numpy.core.multiarray failed to import

这是因为numpy的版本需要更新，直接使用pip更新numpy：

pip install numpy

至此，PyTorch安装成功：

最后，验证GPU是否可用：

返回True，表明已经可以使用GPU了！

参考链接：

linux下 /boot 分区空间不足及其衍生问题

ubuntu16.04 安装NVIDIA和CUDA8.0

ubuntu安装显卡驱动后，右侧菜单及窗体最大最小化消失问题

Ubuntu 17.04 安装 pytorch and pytorchvision

Pytorch如何安装，Linux安装Pytorch，OSX安装Pytorch教程

ImportError: numpy.core.multiarray failed to import

我的博客即将同步至腾讯云+社区，邀请大家一同入驻。