Ubuntu系统搭建深度学习开发环境

正所谓“工欲善其事，必先利其器”。今天分享给大家如何在Ubuntu主机上配置自己的深度学习开发环境。

环境配置

1. Ubuntu 16.04
2. Nvidia CUDA Toolkit 9.0
3. cuDNN V7.1.4 for CUDA 9.0
4. TensorFlow 1.9

安装CUDA Toolkit

第一部分：安装前的准备

1.1 验证你有一个可用CUDA的GPU

$ lspci | grep -i nvidia

1.2 验证你的Linux版本支持CUDA
Nvidia发布的支持CUDA开发工具的Linux发行版本如下表所示。

可以执行如下命令查看系统信息

$ uname -m && cat /etc/*release

1.3 验证你是否安装了gcc

$ gcc --version

1.4 验证系统已经安装了正确的kernel headers和development packages

$ uname -r # 查看内核
$ sudo apt-get install linux-headers-$(uname -r) # 为当前核安装kernel headers和development packages

第二部分：禁用Nouveau驱动

2.1 使用以下命令查看Nouveau