在 Linux 上安装 TensorFlow

TensorFlow是一个流行的深度学习框架，可在各种操作系统上使用。在本文中，我们将介绍如何在Linux上安装TensorFlow框架。

步骤1: 安装pip

首先，在Linux上安装TensorFlow需要安装pip(包管理器)。如果您的系统没有安装pip，请使用以下命令来安装pip。

Ubuntu/Debian

$ sudo apt-get update
$ sudo apt-get install python3-pip

CentOS/Fedora

$ sudo yum install epel-release
$ sudo yum install python3-pip

步骤2: 安装TensorFlow

一旦pip安装完成，我们可以使用pip命令来安装TensorFlow。但是，TensorFlow有两个版本，一个是基于CPU版本的，另一个是基于GPU版本的。如果您的机器具有NVIDIA GPU，则建议安装基于GPU的版本，因为它比基于CPU的版本要快。

安装基于CPU版本

$ sudo pip3 install tensorflow

安装基于GPU版本

首先，您需要安装CUDA和cuDNN，这是NVIDIA GPU加速计算的两个必需库。请根据您的GPU和CUDA版本下载和安装相应的CUDA和cuDNN版本。

然后，执行以下命令安装基于GPU的TensorFlow:

$ sudo pip3 install tensorflow-gpu

步骤3: 验证安装

安装完成后，我们可以通过下面的Python代码片段来测试TensorFlow是否安装成功。

import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))

运行上面的代码，您应该可以看到以下输出:

b'Hello, TensorFlow!'

结论

通过上述步骤，我们成功安装了TensorFlow框架并验证了是否安装成功。现在，您可以开始构建深度学习模型并在Linux上运行它们了。