TensorFlow 是一个强大的机器学习框架,以下是几种常见的安装方式:

1. 使用 pip 安装 📦

pip install tensorflow

✅ 适用于大多数 Python 环境,支持 CPU 和 GPU 版本。
📌 注意:确保已安装 Python 3.7-3.11,并通过 Python 官方文档 确认版本兼容性。

TensorFlow_Installation

2. 从源码编译 🧑‍💻

git clone https://github.com/tensorflow/tensorflow.git
cd tensorflow
./configure
bazel build --config=opt --config=cuda //tensorflow/tools/pip_package:build_pip_package
bazel-bin/tensorflow/tools/pip_package/build_pip_package /tmp/tensorflow_pkg
pip install /tmp/tensorflow_pkg/tensorflow-*.whl

🔍 适合需要自定义配置的用户,详细步骤可参考 TensorFlow 源码编译教程

3. 使用 Docker 🐳

docker run --rm -it tensorflow/tensorflow:latest

🌐 一键启动预配置环境,适合快速测试。更多 Docker 镜像信息请查看 Docker 官方仓库

Docker_Container

4. 其他方式

如需验证安装是否成功,可运行以下代码:

import tensorflow as tf
print(tf.__version__)

✅ 若输出版本号,则表示安装成功!

返回 TensorFlow 主页 了解更多使用技巧。