[ubuntu16.04]cuda+tensorflow安装

第一步 安装cuda和cuDNN：

下载好libcudnn6_6.0.21-1+cuda8.0_amd64.deb

运行脚本sudo ./start.sh

CUDA_REPO_PKG=cuda-repo-ubuntu1604_8.0.61-1_amd64.deb
wget -O /tmp/${CUDA_REPO_PKG} http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/${CUDA_REPO_PKG}
dpkg -i /tmp/${CUDA_REPO_PKG}
rm -f /tmp/${CUDA_REPO_PKG}
dpkg -i libcudnn6_6.0.21-1+cuda8.0_amd64.deb
apt-get update
apt-get install cuda-drivers
apt-get install cuda=8.0.61-1

需要将cuda加入环境变量：sudo vi /etc/profile，把下两行加到/etc/profile末尾

export PATH=/usr/local/cuda-8.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH

重启或者执行以下命令使环境变量生效：

source /etc/profile

第二步 安装tensorflow：

准备 TensorFlow 依赖项和所需的程序包：

sudo apt-get install libcupti-dev

tensorflow-gpu需要python3，所以安装python3

sudo add-apt-repository ppa:jonathonf/python-3.6

sudo apt-get update

sudo apt-get install python3.6

sudo update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.5 1

sudo update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.6 2

sudo update-alternatives --config python3

sudo rm /usr/bin/python3

sudo ln -s python3.5 /usr/bin/python3

然后还要安装pip3，因为pip安装tensorflow-gpu只要一行！

sudo apt install python3-pip

不要直接安装tensorflow-gpu，因为这样版本会不适配。我们要指定版本

pip3 install tensorflow-gpu==1.3.0

第三步 验证安装：

进入python终端以后输入以下代码

import tensorflow as tf
sess = tf.Session()

运行没有错误的话，就会显示机器的GPU信息

 

 

以上三个步骤第二次安装也亲测有效：

不过在我自己的小笔记本上有一个问题，是在最后一步的时候

Could not install packages due to an EnvironmentError: [Errno 13] Permission denied: '__init__.py'
Consider using the `--user` option or check the permissions.

解决方法：

pip3 install tensorflow-gpu==1.3.0 –user

然而我pip3装到了pip3.5里面QAQ

tensorflow-gpu 1.3.0似乎有点老，我改成了1.4.0 仍然匹配cuda8.0和cudnn6。