Ubuntu安装GPU版Tensorflow步骤

此篇前几步配置与我的另一篇博客类似：Ubuntu16.04 cuda8.0 caffe编译安装教程
cuda10.1安装参考：Ubuntu18.04下安装Nvidia驱动和CUDA10.1＋CUDNN
这一篇我在自己电脑上配通了GPU版Tensorflow。
其他参考：Ubuntu16.04下安装Tensorflow-gpu

1、首先禁止系统原有的驱动，然后根据自己的显卡型号在nvidia官网查询对应的驱动

2、然后在命令行模式下（tty）（ctrl+alt+F1-F5）将nouveau加入黑名单禁止启动：

(1)编辑 /etc/modprobe.d/blacklist-nouveau.conf（有的博客说在/etc/modprobe.d/blacklist.conf），增加内容：

blacklist nouveau
options nouveau modest=0

(2) sudo update-initramfs -u

3、安装nvidia显卡驱动
查看自己电脑的对应驱动：https://www.geforce.cn/drivers

sudo /etc/init.d/lightdm stop （关闭图像界面）
sudo apt-get purge nvidia-*   （列出所有的已安装的显卡驱动）
sudo apt-get remove --purge nvidia*（删除所有的显卡驱动）
sudo add-apt-repository ppa:graphics-drivers /ppa
sudo apt-get update
sudo apt-get install nvidia-384（名称根据自己的显卡型号对应的驱动来）
reboot

4、降级gcc（cuda8.0需要，其他可参考对应关系）
(1) 查看当前的gcc版本：gcc --version
(2) 查看已安装的gcc版本： ls /usr/bin/gcc*
(3) 如果版本中没有4.8版本，则安装：sudo apt-get install gcc-4.8
(4) 将4.8版本的优先级提高： sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-4.8 100
(5) 再次查看gcc默认版本是否为4.8,如果是则表示降级成功：gcc --version

5、安装cuda8.0
https://www.cnblogs.com/wmxfd/p/installation_of_nvidia_graphics_driver_and_cuda8_and_cudnn6.htmlsu
(1) 下载安装文件
(2) chmod a+x cuda_8.0.61_375.26_linux.run 设置运行权限
sudo ./cuda_8.0.61_375.26_linux.run.26_linux-run
安装过程中基本上选择默认路径以及选Yes，但是提示是否安装Nvidia驱动时一定选择No， 因为此前已经安装了相应的显卡驱动

(3) 设置环境变量
sudo gedit /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  

(4) 设置动态链接库
终端输入：sudo gedit /etc/ld.so.conf.d/cuda.conf
添加：/usr/local/cuda/lib64
执行生效：sudo ldconfig

(5) 测试cuda的samples

cd /usr/local/cuda/samples/1_Utilities/deviceQuery
make
sudo ./deviceQuery

6、安装cudnn6.0
https://www.cnblogs.com/wmxfd/p/installation_of_nvidia_graphics_driver_and_cuda8_and_cudnn6.html
(1) 去官网注册并根据安装的cuda版本下载对应的库文件，进行解压，官网：
https://developer.nvidia.com/rdp/cudnn-download
(2) 进入include文件夹，执行 sudo cp cudnn.h /usr/local/cuda-8.0/include/ #复制头文件
(3) 进入lib64文件夹，执行 sudo cp lib* /usr/local/cuda/lib64/ #复制动态链接库
(4) 执行

cd /usr/local/cuda/lib64/
sudo rm -rf libcudnn.so libcudnn.so.6 #删除原有动态文件
sudo ln -s libcudnn.so.6.0.21 libcudnn.so.6 #生成软链接
sudo ln -s libcudnn.so.6 libcudnn.so #生成软链接
sudo ldconfig 执行生效

7、安装tensorflow
(1) 下载tensorflow安装包
(2) 安装pip3

apt-get install python3-pip

(3) 安装tensorflow

pip3 install tensorflow_gpu-1.4.0-cp35-cp35m-manylinux1_x86_64.whl