TensorFlow是由Google开发的一个强大开源库，广泛应用于机器学习和深度学习任务，研究人员、开发人员和数据科学家常用它来构建和训练机器学习模型。
如果您正在使用 Ubuntu 系统并希望安装 TensorFlow，本指南将指导您完成在 Ubuntu 上安装和使用 TensorFlow 的过程。
前提条件
在安装 TensorFlow 之前，确保您的 Ubuntu 系统满足以下要求：
- Python: TensorFlow 支持 Python 3.7 到 3.10。
- Pip: Pip 是 Python 的包管理工具。
- 硬件: 虽然 TensorFlow 可以在 CPU 上运行，但拥有 GPU 可以显著加快大模型的训练速度。
步骤 1: 在 Ubuntu 上安装 Python 和 Pip
在安装任何新软件之前，建议您更新包列表并升级已安装的软件包。
sudo apt update
sudo apt upgrade
接下来，安装 Python 和 Pip，Pip 是用于安装和管理 Python 包的包管理工具。
sudo apt install python3
sudo apt install python3-pip
安装完成后，您可以通过以下命令验证 Python 和 Pip 是否安装成功：
python3 --version
pip3 --version
步骤 2: 在 Ubuntu 上创建虚拟环境
创建虚拟环境是可选的，但强烈推荐，因为它有助于将 TensorFlow 安装与其他 Python 项目隔离开来。
为 TensorFlow 项目创建一个新目录，并进入该目录：
mkdir tensorflow_project
cd tensorflow_project
创建虚拟环境并激活它：
python3 -m venv tensorflow_env
source tensorflow_env/bin/activate
激活后，您的终端提示符应该显示虚拟环境的名称，表示虚拟环境已成功激活。
步骤 3: 在 Ubuntu 上安装 TensorFlow
现在您的环境已设置完毕，可以使用 Pip 安装 TensorFlow 的最新稳定版本及其依赖项。
首先升级 Pip：
pip install --upgrade pip
然后安装 TensorFlow：
pip install tensorflow
步骤 4: 验证 TensorFlow 安装
安装完成后，最好验证 TensorFlow 是否已正确安装。您可以进入 Python shell 进行验证：
python3
在 Python shell 中，导入 TensorFlow 并检查其版本：
import tensorflow as tf
print(tf.__version__)
如果 TensorFlow 安装正确，您将看到版本号，类似于：
2.18.0
您还可以运行一个简单的测试来确保 TensorFlow 正常工作：
hello = tf.constant('Hello, TensorFlow!')
print(hello)
输出应该类似于：
tf.Tensor(b'Hello, TensorFlow!', shape=(), dtype=string)
步骤 5: 安装支持 GPU 的 TensorFlow（可选）
如果您拥有兼容的 NVIDIA GPU，并希望利用它进行更快的计算，可以安装支持 GPU 的 TensorFlow。
首先，检查是否有 NVIDIA GPU：
lspci | grep -i nvidia
然后安装 NVIDIA 驱动：
sudo apt install nvidia-driver-535 -y
sudo reboot
接下来，安装支持 GPU 的 TensorFlow：
pip install tensorflow-gpu
最后，通过检查 TensorFlow 是否能检测到您的 GPU 来验证安装是否成功：
import tensorflow as tf
print("Num GPUs Available: ", len(tf.config.experimental.list_physical_devices('GPU')))
您已经成功在Ubuntu 24.04上安装了TensorFlow。无论是进行简单的机器学习项目，还是构建复杂的深度学习模型，TensorFlow都提供了所需的工具。
记得使用虚拟环境来保持您的工作环境整洁，如果需要更快的计算速度，可以考虑启用GPU支持。
