文章目录
众所周知Nvidia-P100是顶级的深度学习神器，价格也是非常昂贵的，将近$8000，有此神器自然是要用起来，所以今天就来为它安装驱动，这是它安装完驱动，并满载工作的样子
Fri Dec 14 08:35:44 2018
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 390.87 Driver Version: 390.87 |
|-------------------------------+----------------------+----------------------+
| GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. |
|===============================+======================+======================|
| 0 Tesla P100-PCIE... Off | 00000000:00:08.0 Off | 0 |
| N/A 38C P0 113W / 250W | 15611MiB / 16280MiB | 67% Default |
+-------------------------------+----------------------+----------------------+
+-----------------------------------------------------------------------------+
| Processes: GPU Memory |
| GPU PID Type Process name Usage |
|=============================================================================|
| 0 1421 C python 15601MiB |
+-----------------------------------------------------------------------------+
环境
- ubuntu 16.04
- Nvidia-P100
调研
去Nvidia官网以及各大博客论坛，最终决定安装nvidia-390这个驱动，原因有如下：
- 驱动的版本较新
- 很多程序都是用较新版本的驱动做编译的，如果使用旧版本在启动tensorflow-gpu加速时会遇到问题，我就曾经遇到过
- 此驱动较稳定，是比较合适的选择
卸载之前安装的旧驱动
如果之前因为没有调研清楚就安装了其他驱动，需要将其卸载干净，直接用apt卸载
sudo apt-get purge nvidia*
用PPA安装
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt update
sudo apt install nvidia-390
确认驱动
lsmod | grep nvidia
重启机器
reboot
输入指令
nvidia-smi
看到如下，说明安装成功