TensorFlow-gpu 在 CentOS 7 下的完全安装手册
鉴于目前部分TensorFlow的Installation Guide已经有些落后于最新版本,以下我将结合我在CentOS下安装时踩过的坑以及 官方教程 ,介绍一下如何在CentOS 7下安装支持GPU运算的TensorFlow 1.4.1。
在安装过程中切记,一定要选择正确的版本进行安装!
I. 检查系统环境
在安装之前,需要先检查系统的软硬件环境是否支持CUDA与TF的安装,具体来说,主要有以下几个步骤:
- 检查是否电脑配置有Nvidia显卡
$ /usr/sbin/lspci | grep -i nvidia
执行结果如下:
- 检查是否安装了正确的GCC版本
$ gcc --version
执行结果如下:
II. 安装 CUDA 与 cuDNN
完成了系统环境的检查,就可以开始安装CUDA与cuDNN了。这一步是安装带有GPU支持的TensorFlow必须完成的,否则将无法使用GPU来完成机器学习任务。安装CUDA也是坑最多的一步,没有之一,基本上在安装上遇到的问题基本上都集中在这里,所以一定要谨慎操作。具体的步骤如下:
1. 确定CUDA与cudnn的版本
请参照文章开头的表格,选择与Tensor