TensorFlow-gpu 在 CentOS 7 下的完全安装手册

TensorFlow-gpu 在 CentOS 7 下的完全安装手册

鉴于目前部分TensorFlow的Installation Guide已经有些落后于最新版本，以下我将结合我在CentOS下安装时踩过的坑以及 官方教程 ，介绍一下如何在CentOS 7下安装支持GPU运算的TensorFlow 1.4.1。

在安装过程中切记，一定要选择正确的版本进行安装！

I. 检查系统环境

在安装之前，需要先检查系统的软硬件环境是否支持CUDA与TF的安装，具体来说，主要有以下几个步骤：

1. 检查是否电脑配置有Nvidia显卡

$ /usr/sbin/lspci | grep -i nvidia

执行结果如下：

1. 检查是否安装了正确的GCC版本

$ gcc --version

执行结果如下：

II. 安装 CUDA 与 cuDNN

完成了系统环境的检查，就可以开始安装CUDA与cuDNN了。这一步是安装带有GPU支持的TensorFlow必须完成的，否则将无法使用GPU来完成机器学习任务。安装CUDA也是坑最多的一步，没有之一，基本上在安装上遇到的问题基本上都集中在这里，所以一定要谨慎操作。具体的步骤如下：

1. 确定CUDA与cudnn的版本

请参照文章开头的表格，选择与Tensor