TensorFlow 是一个开源机器学习框架,支持多种平台和硬件配置。以下是详细的安装步骤,帮助您快速开始:


1. 确定环境需求 📋

  • 操作系统:支持 Windows、Linux、macOS

    Windows
    Linux
    macOS
  • Python 版本:推荐使用 Python 3.8-3.11

    Python
  • 硬件要求

    • CPU 安装:无需额外配置
    • GPU 加速:需安装 NVIDIA 驱动和 CUDA 工具包(推荐参考 GPU 配置指南

2. 安装步骤 🛠

2.1 通过 pip 安装

pip install tensorflow

⚠️ 若需 GPU 支持,请安装 TensorFlow GPU 版本

2.2 通过 Docker 安装

docker run -it --rm gcr.io/tensorflow/tensorflow:latest

🌐 更多 Docker 配置详情请访问 Docker 部署文档

2.3 从源码编译(高级用户)

git clone https://github.com/tensorflow/tensorflow.git
cd tensorflow
./configure
bazel build --config=opt --config=cuda //tensorflow:libtensorflow.so

📚 源码编译教程可参考 官方开发者指南


3. 验证安装 ✅

运行以下代码检查 TensorFlow 是否正常:

import tensorflow as tf
print(tf.__version__)

🎉 若输出版本号,说明安装成功!如需进一步测试,可访问 TensorFlow 官方示例


4. 常见问题解答 ❓

  • Q: 安装后无法使用 GPU?
    A: 请确保已正确安装 CUDA 和 cuDNN,参考 GPU 配置指南

  • Q: 如何升级 TensorFlow?
    A: 使用 pip install --upgrade tensorflow 即可。


📦 更多 TensorFlow 资源 | 📘 官方文档 | 🛠 安装工具包下载