Ubuntu 16.04安装NVIDIA GTX 1070和TensorFlow指南

0. 准备工作
硬件：NVIDIA GTX 1070显卡一块
软件：Ubuntu 16.04.1/CUDA 8.0 RC/cuDNN 5.1/TensorFlow 0.10.0
1. 安装Ubuntu 16.04.1
2. 安装NVIDIA 365.35 驱动程序

1 2 3 4

sudo add-apt-repository ppa:graphics-drivers/ppa sudo apt-get update sudo apt-get install nvidia-367 sudo reboot

3. 安装CUDA 8.0 RC，需要使用runfile安装方案，不要选择deb方案。

1	sudo sh cuda_8.0.27_linux.run

备注：在安装过程中非常关键的地方是会提示你是否安装361这个低版本的驱动：

1	Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 361.62?

答案必须是n，因为之前我们已经安装过高版本的GTX1070驱动了。
4. 使用nvidia-smi命令测试显卡驱动和CUDA是否安装成功

1 2 3 4 5 6 7 8 9 10

Thu Aug 4 01:19:40 2016 +-----------------------------------------------------------------------------+ | NVIDIA-SMI 367.35 Driver Version: 367.35 | |-------------------------------+----------------------+----------------------+ | GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC | | Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. | |===============================+======================+======================| | 0 GeForce GTX 1070 Off | 0000:01:00.0 Off | N/A | | 0% 40C P8 11W / 166W | 103MiB / 8112MiB | 0% Default | +-------------------------------+----------------------+----------------------+

5. 安装cuDNN 5.1

1 2 3 4

tar -zxf cudnn-8.0-linux-x64-v5.1.tgz cd cuda sudo cp lib64/* /usr/local/cuda/lib64/ sudo cp include/* /usr/local/cuda/include/

6. 在.bashrc文件中添加和CUDA相关的环境变量

1 2	export PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}} export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

7. 然后在命令行输入

1	source ~/.bashrc

8. 使用命令nvcc -V测试环境变量是否配置成功

1 2 3 4 5

$ nvcc -V nvcc: NVIDIA (R) Cuda compiler driver Copyright (c) 2005-2016 NVIDIA Corporation Built on Sun_Sep__4_22:14:01_CDT_2016 Cuda compilation tools, release 8.0, V8.0.44

9. 过源代码方式编译安装TensorFlow GPU版本