Ubuntu16.04环境下PyTorch简易安装教程
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了！
参考链接：
ubuntu安装显卡驱动后，右侧菜单及窗体最大最小化消失问题
Ubuntu 17.04 安装 pytorch and pytorchvision