在CentOS 7上安装Tensorflow

最新推荐文章于 2022-07-20 19:22:21 发布

iamMorgen

最新推荐文章于 2022-07-20 19:22:21 发布

阅读量2.9k

点赞数

CC 4.0 BY-SA版权

分类专栏：深度学习文章标签： tensorflow centos

本文链接：https://blog.youkuaiyun.com/u013054973/article/details/78549676

本文详细介绍了如何在CentOS 7系统中安装TensorFlow，包括GPU驱动、CUDA、cuDNN的安装，以及Python环境和TensorFlow的配置。此外，还提供了TensorFlow的基本使用示例，如模块导入和MNIST数据集的训练。

简介

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:/usr/local/cuda-8.0/extras/CUPTI/lib64:$LD_LIBRARY_PATH
export CUDA_HOME=/usr/local/cuda-8.0/

最低0.47元/天解锁文章

新学期VIP享超值加赠