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

https://blog.csdn.net/red_stone1/article/details/78727096

 版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/red_stone1/article/details/78727096
安装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
1
卸载可能存在的旧版本NVIDIA 驱动（这一步是必须的，不然重启会有问题）:

sudo apt-get remove --purge nvidia*
1
安装驱动可能需要的依赖：

sudo apt-get update
sudo apt-get install dkms build-essential linux-headers-generic
1
2
把 nouveau 驱动加入黑名单并禁用用 nouveau 内核模块（如果新重装的系统，没有驱动起作用，忽略）：

sudo nano /etc/modprobe.d/blacklist-nouveau.conf
1
在文件 blacklist-nouveau.conf 中加入如下内容：

blacklist nouveau 
options nouveau modeset=0

保存退出，执行：

sudo update-initramfs -u
1
然后重启：

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

sudo service lightdm stop
1
进入之前NVIDIA驱动文件下载目录，安装驱动：

sudo chmod u+x NVIDIA-Linux-x86_64-384.98.run
sudo ./NVIDIA-Linux-x86_64-384.98.run -no-opengl-files
1
2
-no-opengl-files表示只安装驱动文件，不安装OpenGL文件。这个参数不可忽略，否则会导致登录界面死循环。

最后重新启动图形环境：

sudo service lightdm start
1
可以通过以下命令确认驱动是否正确安装：

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

首先安装compiz配置管理器：

sudo apt-get install compizconfig-settings-manager
1
然后，打开配置管理器：

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

接着，重置compiz：

dconf reset -f /org/compiz/
1
最后注销重启：

gnome-session-quit
1
至此，NVIDIA显卡驱动安装成功。

科学上网的简易方法
进入PyTorch官网： 
PyTorch

显示如下：

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

conda install pytorch torchvision cuda80 -c soumith
1
其中，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
1
安装torchvision
安装torchvision比较简单，可直接使用pip命令安装：

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

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

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

ImportError: numpy.core.multiarray failed to import
1
这是因为numpy的版本需要更新，直接使用pip更新numpy：

pip install numpy
1
至此，PyTorch安装成功：

最后，验证GPU是否可用：

返回True，表明已经可以使用GPU了！
--------------------- 
作者：红色石头Will 
来源：CSDN 
原文：https://blog.csdn.net/red_stone1/article/details/78727096 
版权声明：本文为博主原创文章，转载请附上博文链接！