Ubuntu16.04系统安装tensorflow(GPU)

作者:冯拓

电脑配置如下：

配置	HP-Z820
CPU核心线程数和主频	intel xeon（至强） E-5  2620  2.0GHz*24
内存	64GB
硬盘	2TB
显卡	NIVDIA TITAN X 12GB

安装过程中使用的安装包：

	安装包
驱动	NVIDIA-Linux-x86_64-396.18.run
cuda	cuda_9.1.85_387.26_linux.run
cudnn	cudnn-9.1-linux-x64-v7.1.tgz
Anaconda	Anaconda3-5.2.0-Linux-x86_64.sh
bazel	bazel-0.14.1-installer-linux-x86_64.sh
tensorflow源代码	tensorflow-r1.8
cudnn	cudnn-9.1-linux-x64-v7.1.tgz

cuda下载链接：

https://developer.nvidia.com/cuda-downloads

anaconda下载链接：

https://www.continuum.io/downloads 

bazel下载链接：

https://github.com/bazelbuild/bazel/releases

tensorflow源代码下载链接：
https://github.com/tensorflow/tensorflow

---

本文分为两个大部分来介绍，首先介绍cuda与cudnn的安装，然后介绍使用源码安装tensorflow。

 

一、安装cuda与cudnn

1、Nvidia驱动安装-run文件安装

下载完名称为NVIDIA-Linux-x86_64-396.18.run 的文件。使用以下命令：

sudo gedit /etc/modprobe.d/blacklist.conf

加入以下语句，将nouveau禁止命令写入文件。

blacklist nouveau 
blacklist lbm-nouveau 
options nouveau modeset=0 
alias nouveau off 
alias lbm-nouveau off

调用指令禁止nouveau。

echo options nouveau modeset=0 | sudo tee -a /etc/modprobe.d/nouveau-kms.conf

建立新的内核并重启。

sudo update-initramfs -u
sudo reboot

使用ctrl + alt+ F1进入文本模式，关闭x server。