服务器上配置Tensorflow GPU版

服务器上配置Tensorflow GPU版

第一次写博客，新手一枚，CSDN之路，多多指教。

一、主要参考

1）（超完整）Linux（debian9）服务器配置tensorflow环境：nvidia驱动、CUDA、cudnn、anaconda
http://blog.csdn.net/Star_code/article/details/76616958

2）阿里云centOS6 下python安装及配置、pip安装及配置、ipython安装及配置
http://www.cnblogs.com/Vdiao/p/6262513.html

3）”libcudnn.so.5 cannot open shared object file: No such file or directory”
http://blog.csdn.net/u014696921/article/details/60140264

4）libcudnn.so.6 –csdn下载
http://download.csdn.net/download/guotong1988/9918527

二、配置介绍

• Ubuntu 16.04.3（系统版本）

• NVIDIA-SMI 384.90 （硬件驱动）
  可以用如下命令查看驱动

  cat /proc/driver/nvidia/version

• TITAN Xp（GPU型号）

三、安装流程

• Cuda8.0
• Cudnn6.0（此处有坑）
• Anaconda3（可选）
• Tensoflow-GPU

四、详细流程

Note：因为我服务器本身就是root 账户，所以不需要sudo 指令。

1）CUDA的安装

直接安装CUDA toolkit，官网如下，tensoflow支持cuda8.0
https://developer.nvidia.com/cuda-downloads
cuda_8.0.61_375.26_linux.run（文件名）

sh cuda_8.0.61_375.26_linux.run

按照提示一步一步来，不用安驱动

安装结束后添加环境变量（我习惯用vim）

vim /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

使更改生效。

nvcc –V

检查CUDA，如果显示如下，则CUDA安装完成

2）CUDNN的安装

要下载合适的版本

https://developer.nvidia.com/rdp/cudnn-download （官网）
cudnn-8.0-linux-x64-v5.1.tgz（不兼容）

解决方案：
cudnn-8.0-linux-x64-v6.0.tgz（兼容）
http://download.csdn.net/download/guotong1988/9918527

或者在百度云盘下载，链接：http://pan.baidu.com/s/1dFs8o4D 密码：a0da

tar -xvzf cudnn-8.0-linux-x64-v6.0.tgz

解压后是cuda文件夹

cd cuda

cp include/cudnn.h /usr/local/cuda/include

cp lib64/libcudnn.（版本eg: so.6） /usr/local/cuda/lib64

CUDNN安装完成

3）Anaconda3安装（可选）

https://www.continuum.io/downloads（官网，比较慢）

例如这个版本
Anaconda3-5.0.0.1-Linux-x86_64.sh

sh Anaconda3-5.0.0.1-Linux-x86_64.sh

进行安装，安装目录为

/home/userName/anaconda3

最后出现是否添加环境变量选择yes，如果选择了no，可以之后手动添加，方法如下

vim ~/.bashrc

在最后添加

export PATH=$PATH:/home/userName/anaconda3/bin:$PATH

source ~/.bashrc

使其生效（ps.自动添加也需要source这一步）

4）安装tensoflow

pip install tensorflow-gpu

如果中途失败，或者下载速度不够块，换源，不会的小伙伴可以参考这篇文章
http://blog.csdn.net/qq_32897957/article/details/53023738

没有安装pip的可以参考这篇文章
http://www.cnblogs.com/Vdiao/p/6262513.html

5）测试

pythonimport tensorflow

如果之前安装的cudnn文件是cudnn-8.0-linux-x64-v5.1.tgz，则会出现以下问题：

核心部分是

解决方法就是安装这个版本的cudnn，cudnn-8.0-linux-x64-v6.0.tgz

测试样例

$ python...>>> import tensorflow as tf>>> hello = tf.constant('Hello, TensorFlow!')>>> sess = tf.Session()>>> print(sess.run(hello))Hello, TensorFlow!>>> a = tf.constant(10)>>> b = tf.constant(32)>>> print(sess.run(a + b))42>>>

结果如下代表配置成功，大功告成