Tensorflow-gpu版本安装

目录

• 服务器选型
• NVIDIA GPU驱动安装
• cuda和cudnn的安装
  • cuda安装
  • cudnn的安装
• tensorflow-gpu安装

最近给公司部署一套深度学习相关的环境，以tensorflow为框架。简单整理下整个的部署过程。

tensorflow官方网站：https://www.tensorflow.org

服务器选型

我们的tensorflow是基于gpu的版本，使用的是tensorflow-gpu 1.12.0版本。既然是gpu的版本，那么首先得需要一个带有gpu的服务器。我们这里直接使用阿里云服务器，型号如下：

NVIDIA GPU驱动安装

那么既然有了带gpu的服务器，gpu驱动就必然是一个绕不开的话题。通过上面的选型我们可以知道，阿里云的购买页面明确给出了gpu的型号，我们需要去到gpu官方网站，根据型号选择驱动。

通过如下页面查找相应的GPU驱动：

https://www.nvidia.com/Download/index.aspx?lang=en-us

下载的驱动名称示例：NVIDIA-Linux-x86_64-410.104.run

安装的话相对比较简单：

# 通过shell运行，然后按照提示操作即可
sh NVIDIA-Linux-x86_64-410.104.run

验证是否正常安装了：

[root@ctnr ~]# nvidia-smi
Tue Mar 26 11:31:34 2019
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 410.104      Driver Version: 410.104      CUDA Version: 10.0     |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|===============================+======================+======================|
|   0  Tesla P4            Off  | 00000000:00:08.0 Off |                    0 |
| N/A   28C    P8     7W /  75W |      0MiB /  7611MiB |      0%      Default |
+-------------------------------+----------------------+----------------------+

+-----------------------------------------------------------------------------+
| Processes:                                                       GPU Memory |
|  GPU       PID   Type   Process name                             Usage      |
|=============================================================================|
|  No running processes found                                                 |
+-----------------------------------------------------------------------------+

查看当前服务器显卡的个数：

yum install -y lspci

[root@ctnr ~]# lspci|grep -i nvidia
00:08.0 3D controller: NVIDIA Corporation GP104GL [Tesla P4] (rev a1)

其中00:08.0是显卡的代号，通过如下方式可以查看详细的显卡信息：

[root@ctnr ~]# lspci -v -s 00:08.0
00:08.0 3D controller: NVIDIA Corporation GP104GL [Tesla P4] (rev a1)
	Subsystem: NVIDIA Corporation Device 11d8
	Physical Slot: 8
	Flags: bus master, fast devsel, latency 0, IRQ 39
	Memory at fd000000 (32-bit, non-prefetchable) [size=16M]
	Memory at e0000000 (64-bit, prefetchable) [size=256M]
	Memory at f2000000 (64-bit, prefetchable) [size=32M]
	Capabilities: [60] Power Management version 3
	Capabilities: [68] MSI: Enable+ Count=1/1 Maskable- 64bit+
	Capabilities: [78] Express Endpoint, MSI 00
	Capabilities: [c8] Vendor Specific Information: Len=08 <?>
	Kernel driver in use: nvidia
	Kernel modules: nouveau, nvidia_drm, nvidia

这里是tensorflow官方给出的gpu支持：https://www.tensorflow.org/install/gpu

cuda和cudnn的安装

tensorflow-gpu要想正常运行，除了必要的gpu驱动，还依赖cuda和cudnn两个sdk。

下面是tensorflow-gpu版本依赖的cuda和cudnn的版本：

https://www.tensorflow.org/install/source

cuda是深度学习的sdk

cudnn是神经网络的sdk

cuda安装

cuda的下载地址：

https://developer.nvidia.com/cuda-90-download-archive

需要下载cuda包，以及所有的补丁文件，我这里因为是centos，所以下载的全是rpm包。直接通过yum安装即可：

yum localinstall -y cuda-repo-rhel7-9-0-*.rpm

需要说明的是，这些包都只是本地仓库包，也就是说，安装了这些包，并不会真正安装cuda，而只是把cuda的包在本地生成了一个本地yum源。这个时候，就可以直接使用yum install -y cuda来安装cuda了。

验证安装：

[root@ctnr ~]# cat /usr/local/cuda/version.txt
CUDA Version 9.0.176

cudnn的安装

cudnn的下载地址：

https://developer.nvidia.com/rdp/cudnn-download

需要说明的是，要下载cudnn，需要先登录，这就要求我们得要有nvidia的帐号。

这两个包也是rpm包，直接安装即可。

安装完成后的验证方式如下：

[root@ctnr ~]# cat /usr/include/cudnn_v7.h |grep CUDNN_MAJOR -A 2
#define CUDNN_MAJOR 7
#define CUDNN_MINOR 5
#define CUDNN_PATCHLEVEL 0
--
#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)

#include "driver_types.h"

上面的三行，分别代表主版本、次版本以及修定版本

在上图中，其实可以看到，除了定制的系统版本，cudnn也提供通用版本，即cnDNN Library for Linux，下载下来是一个tar.gz的文件，我们可以通过如下方式安装即可：

tar xf cudnn-9.0-linux-x64-v7.5.0.56.tgz
cp cuda/lib/64/libcudnn.so* /usr/local/cuda/lib64/
cp cuda/lib64/libcudnn_static.a /usr/local/cuda/lib64/
cp cuda/include/cudnn.h /usr/include/

另外，还需要说明的是，我尝试在windows上直接下载通用版本的时候，下载的总是一个solitairetheme8为后缀的一个卡牌游戏。在linux上通过wget的方式下载，则是正常的压缩包。

tensorflow-gpu安装

以上配置都做好了以后，tensorflow-gpu的安装相对倒是比较简单。官方基于Bazel使用源码编译的安装方式可以直接参考这里：https://www.tensorflow.org/install/source#build_the_package

我这里简化部署，直接使用pip安装：

yum install -y python36 python36-pip

pip3 install -U pip six numpy wheel mock
pip3 install -U  keras_applications==1.0.5 --no-deps
pip3 install -U keras_preprocessing==1.0.3 --no-deps

pip3 install tensorflow-gpu-1.12.0

至此，tensorflow安装完成。