ubuntu16.04安装gpu版本的tensorflow

过程参考了http://blog.csdn.net/continueoo/article/details/71404066。

安装CUDA® Toolkit 9.0

下载

从网站https://developer.nvidia.com/cuda-downloads下载合适版本的cuda toolkit。

安装

sudo dpkg -i cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64.deb
sudo apt-key add /var/cuda-repo-9-0-local/7fa2af80.pub
sudo apt update
sudo apt install cuda

修改环境变量

编辑/etc/profile文件，在最后加上下面这一行：

export PATH=/usr/local/cuda-9.0/bin${PATH:+:${PATH}}

验证安装成功

重启之后输入如下命令：

cat /proc/driver/nvidia/version

出现如下内容：

输入如下命令：

nvcc -V

出现如下内容：

安装cuDNN v7

下载

从网站https://developer.nvidia.com/rdp/cudnn-download下载下面要安装的包

安装runtime library

sudo dpkg -i libcudnn7_7.0.4.31-1+cuda9.0_amd64.deb

安装developer library

sudo dpkg -i libcudnn7-dev_7.0.4.31-1+cuda9.0_amd64.deb 

安装code samples和cuDNN Library User Guide

sudo dpkg -i libcudnn7-doc_7.0.4.31-1+cuda9.0_amd64.deb

验证

cp -r /usr/src/cudnn_samples_v7/ $HOME
cd $HOME/cudnn_samples_v7/mnistCUDNN
make clean && make
 ./mnistCUDNN

出现test passed信息。

安装libcupti-dev library

sudo apt install libcupti-dev

安装gpu版本的tensorflow

sudo pip install tensorflow-gpu

结束。