简介
Tensorflow 是Google基于DistBelief进行研发的第二代人工智能学习系统,是采用数据流图(data flow graphs),用于数值计算的开源软件库。架构灵活,可以部署于一块或多块CPU、GPU卡的各种平台。最初用由Google大脑小组开发出来,用于机器学习和深度神经网络方面的研究,系统的通用性使其可以广泛用于其他计算领域,是目前使用最广的深度学习框架之一。
本教程主要介绍TensorFlow 在Centos 7上的安装和使用,包括安装过程,基本使用和简单示例运行。
GPU驱动安装
根据GPU型号从相应网站下载驱动,例如使用NVIDIA Tesla M60,从NVIDIA网站选择对应的型号和操作系统,CUDA Toolkit版本,下载驱动文件,如NVIDIA-Linux-x86_64-375.66.run,运行驱动文件,根据提示安装:
sh NVIDIA-Linux-x86_64-375.66.run
安装完成后可以通过NVIDIA命令工具nvidia-smi查看GPU情况:
nvidia-smi
Wed Jun 28 11:39:28 2017
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 375.39 Driver Version: 375.39 |
|-------------------------------+----------------------+----------------------+
| 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 M60 Off | 0000:00:02.0 Off | Off |
| N/A 36C P0 38W / 150W | 0MiB / 8123MiB | 0% Default |
+-------------------------------+----------------------+----------------------+
+-----------------------------------------------------------------------------+
| Processes: GPU Memory |
| GPU PID Type Process name Usage |
|=============================================================================|
| No running processes found |
+-----------------------------------------------------------------------------+:
TensorFlow 安装过程
1.安装CUDA
从NVIDIA网站选择最新的驱动版本,选择Linux,x86_64,CentOS 7,下载rpm(local)驱动文件,并安装
sudo rpm -i cuda-repo-rhel7-8-0-local-ga2-8.0.61-1.x86_64.rpm
sudo yum clean all
sudo yum install cuda
设置环境变量
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64