Tensorflow的CUDA升级，以及相关配置

本文主要介绍Tensorflow在unbuntu下的cuda升级以及一些相关配置–GPU，cuDNN，Cuda SDK。目的是为了解决安装完成tensorflow之后在使用过程中想调整上述相关设置，比如禁用GPU、升级cuda等。

本文内容要求tensorflow已经安装完成，主要涉及GPU版本相关设置

解决的问题：
1、安装了GPU版本的tensorflow，想禁用tensorflow对GPU的使用
2、升级Cuda SDK与cuDNN，以及设置相关配置

1、安装Cuda SDK

官网下载和安装Cuda Toolkit，选择合适的系统版本下载，本文为介绍基于ubuntu系统的版本，个人建议下载runfile(local)的Installer Type。关于显卡驱动的安装可以参见（链接-安装最新Nvidia显卡驱动 ）

如果下载文件runfile(local)，大小约为1G+，文件名称格式为‘cuda_x.x.xx_linux.run’（其中x为数字，表示版本），按照下载页面安装指导安装即可。

#shell 
sudo Run `sudo sh cuda_x.x.xx_linux.run`

安装过程中会设置安装路径，如果是7.5版本，那么默认安装在路径‘/usr/local/cuda-7.5’,并在‘/usr/local/cuda’做一份映射，此项为可选项目，如果不想覆盖前版，可以不做映射。

安装完成需要添加路径，特别是在安装不同版本的cuda共存情况下。需要在profile或.bashrc中添加一下内容：

#添加cuda路径
PATH=$PATH:/usr/local/cuda-7.5 (CUDA安装路径)
#添加lib路径
LD_LIBRARY_PATH=$LA_LIBRARY_PATH:/usr/local/cuda-7.5/lib64
#-----------#
#如果做了映射，也可以使用以下内容
PATH=$PATH:/usr/local/cuda
#添加lib路径
LD_LIBRARY_PATH=$LA_LIBRARY_PATH:/usr/local/cuda/lib64

2、安装cuDNN

官网下载和安装cuDNN，择合适的系统版本下载，需要注意的是，版本的选择与已经安装的Cuda Toolkit版本有关，否则会报错。

#版本不对可能引发的错误提示
E tensorflow/stream_executor/cuda/cuda_dnn.cc:286] Loaded cudnn library: 5005 but source was compiled against 4007. If using a binary install, upgrade your cudnn library to match. If building from sources, make sure the library loaded matches the version you specified during compile configuration.

下载的文件名为，‘cudnn-#.#-linux-x64-v*.tgz’，大约80M左右，其中#.#为版本号–如‘8.0’，v*为版本–如‘v5’。

tar xvzf cudnn-7.0-linux-x64-v4.tgz
#注意cuda路径，与之前安装路径一致
sudo cp cudnn-7.0-linux-x64-v4/cudnn.h /usr/local/cuda/include
sudo cp cudnn-7.0-linux-x64-v4/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

cuDNN安装完成

3、配置tensorflow

如果tensorflow是使用whl文件安装，需要下载源码进行配置，官网推荐下载地址Tensorflow
或者使用命令下载

#shell
git clone https://github.com/tensorflow/tensorflow

进入下载的tensorflow目录，以root权限运行文件‘configure’

$ sudo ./configure
#开始设置或更改配置
#设置与tensorflow关联的python路径
Please specify the location of python. [Default is /usr/bin/python]:

#设置是否开启GPU支持
Do you wish to build TensorFlow with GPU support? [y/N] y
GPU support will be enabled for TensorFlow

#设置所安装的Cuda SDK版本
Please specify the Cuda SDK version you want to use, e.g. 7.0. [Leave
empty to use system default]: 7.5

#设置关联的cuda sdk安装路径
Please specify the location where CUDA 7.5 toolkit is installed. Refer to
README.md for more details. [default is: /usr/local/cuda]: /usr/local/cuda

#设置cuDNN的版本
Please specify the Cudnn version you want to use. [Leave empty to use system
default]: 7.0

#设置cudnn的库文件所存放的路径，即与之前设置的相同，
#确认后将会根据路径与版本号自动扫描搜索是否有相关库文件
Please specify the location where the cuDNN 7.0 library is installed. Refer to
README.md for more details. [default is: /usr/local/cuda]: /usr/local/cuda-7.5

Please specify a list of comma-separated Cuda compute capabilities you want to
build with. You can find the compute capability of your device at: 
https://developer.nvidia.com/cuda-gpus.
#选择gpu的计算能力，可以去官网查看相应型号GPU的计算能力<https://developer.nvidia.com/cuda-gpus>
Please note that each additional compute capability significantly increases your
build time and binary size. [Default is: \"3.5,5.2\"]: 3.5

Setting up Cuda include
Setting up Cuda lib64
Setting up Cuda bin
Setting up Cuda nvvm
Configuration finished

---

对tensorflow的cuda升级，配置完成

---

---

相关链接：

[1] 安装Tensorflow（Linux ubuntu） http://blog.csdn.net/lenbow/article/details/51203526
[2] ubuntu下CUDA编译的GCC降级安装 http://blog.csdn.net/lenbow/article/details/51596706
[3] ubuntu手动安装最新Nvidia显卡驱动 http://blog.csdn.net/lenbow/article/details/51683783