ubuntu16配置Mask R-CNN深度学习环境——驱动显卡+cuda+pytorch

一、 下载安装nvidia显卡驱动
先去nvidia官网查看对应显卡适配驱动型号

1. 先卸载原有N卡驱动
   #for case1: original driver installed by apt-get:
   sudo apt-get remove --purge nvidia*
   #for case2: original driver installed by runfile:
   sudo chmod +x *.runsudo ./NVIDIA-Linux-x86_64-384.59.run --uninstall

如果原驱动是用apt-get安装的，就用第1种方法卸载。
如果原驱动是用runfile安装的，就用–uninstall命令卸载。其实，用runfile安装的时候也会卸载掉之前的驱动，所以不手动卸载亦可。

2. 禁用nouveau驱动
   sudo gedit /etc/modprobe.d/blacklist.conf
   在文本最后添加：（禁用nouveau第三方驱动，之后也不需要改回来）
   blacklist nouveau
   options nouveau modeset=0
   然后执行：
   sudo update-initramfs -u
   重启后，执行：lsmod | grep nouveau。如果没有屏幕输出，说明禁用nouveau成功。
3. 禁用X-Window服务
   sudo service lightdm stop #这会关闭图形界面，但不用紧张
   按Ctrl-Alt+F1进入命令行界面，输入用户名和密码登录即可。
4. 命令行安装驱动
   #给驱动run文件赋予执行权限：sudo chmod +x NVIDIA-Linux-x86_64-390.87.run大小写键盘问题，Ubuntu不认小键盘，所以一定要关掉NumLock，使用字母键盘上面的数字按键进行输入。#后面的参数非常重要，不可省略：sudo ./NVIDIA-Linux-x86_64-390.87.run –no-opengl-files –no-opengl-files：表示只安装驱动文件，不安装OpenGL文件。这个参数不可省略，否则会导致登陆界面死循环，英语一般称为”login loop”或者”stuck in login”。
   （这是copy别人的，自己的没记住，我也是尝试选择了好多遍才安装好）
   The distribution-provided pre-install script failed! Are you sure you want to continue? 选择 yes 继续。
   Would you like to register the kernel module souces with DKMS? This will allow DKMS to automatically build a new module, if you install a different kernel later? 选择 No 继续。
   问题没记住，选项是：install without signing
   问题大概是：Nvidia’s 32-bit compatibility libraries? 选择No 继续。
   Would you like to run the nvidia-xconfigutility to automatically update your x configuration so that the NVIDIA x driver will be used when you restart x? Any pre-existing x confile will be backed up. 选择 Yes继续

这些选项如果选择错误可能会导致安装失败，没关系，只要前面不出错，多尝试几次就好。
之后，按照提示安装，成功后重启即可。
如果提示安装失败，不要急着重启电脑，重复以上步骤，多安装几次即可。
Driver测试：nvidia-smi #若列出GPU的信息列表，表示驱动安装成功
小提示：在命令行输入：sudo service lightdm start ，然后按Ctrl-Alt+F7即可恢复到图形界面。

二、 安装cuda8.0
sudo ./cuda_8.0.61_375.26_linux.run --no-opengl-libs --no-opengl-libs：表示只安装驱动文件，不安装OpenGL文件。必需参数，原因同上。注意：不是-no-opengl-files。
按照提示安装即可。我依次选择了：

1. accept #同意安装
2. n #不安装Driver，因为已安装最新驱动
3. y #安装CUDA Toolkit
4. #安装到默认目录
5. y #创建安装目录的软链接
6. n #不复制Samples，因为在安装目录下有/samples
   CUDA Sample测试：
   #编译并测试设备 deviceQuery：
   cd /usr/local/cuda-8.0/samples/1_Utilities/deviceQuery
   sudo make
   ./deviceQuery

三、安装pytorch

1. 安装pip
   对于python3: sudo apt-get install python3-pip
   对于python2: sudo apt-get install python-pip
2. 下载安装pytorch

1. 官网下载pytorch 我下载的是：torch-0.2.0.post3-cp35-cp35m-manylinux1_x86_64.whl（0.2.0代表pytorch版本，35代表python版本3.5）
2. 安装pytorch
   a) 进入安装包下载目录
   b) 命令1：sudo pip3 install torch-0.2.0.post3-cp35-cp35m-manylinux1_x86_64.whl(因为使用python3,所意使用命令pip3,否则使用pip。sudo是为给python3.5赋予安装权限
   *【说明如果出现cannot import main form pip3的错误，则进行以下操作：命令：sudo gedit /usr/bin/pip3
   将from pip import main改为from pip._internal import main
   pip上同】
   c) 命令2：sudo pip3 install torchvision
   d) 命令3：python3(如果安装2.5版本则python。此命令用于打开python）
   打开python后输入以下两条命令
   import torch
   import torchvision
   再输入命令 print(torch.cuda.is_available())
   返回ture证明安装成功