欢迎来到ABC Compute Forum的深度学习教程系列!本指南将帮助你快速安装TensorFlow,开启机器学习之旅。🛠️

安装前准备 📋

  1. 系统要求

  2. 选择版本

    • CPU版:pip install tensorflow
    • GPU版:pip install tensorflow-gpu(需先安装CUDA)
    • 最新稳定版:TensorFlow官网

安装步骤 ⬇️

步骤1:创建虚拟环境

python -m venv tf_env
source tf_env/bin/activate  # Linux/macOS
tf_env\Scripts\activate     # Windows
虚拟环境配置

步骤2:安装TensorFlow

pip install tensorflow

安装完成后可通过 python -c "import tensorflow as tf; print(tf.__version__)" 验证版本

步骤3:配置GPU加速(可选)

  • 安装CUDA Toolkit 11.x
  • 安装cuDNN 8.x
  • 配置环境变量:LD_LIBRARY_PATH(Linux)或 PATH(Windows)
TensorFlow_GPU_Configuration

常见问题 ❓

  • Q: 安装失败如何解决?
    A: 检查依赖项安装指南

  • Q: 如何验证安装是否成功?
    A: 运行以下代码:

    import tensorflow as tf
    print("GPU可用:", tf.config.list_physical_devices('GPU'))
    

扩展学习 📚

TensorFlow_Logo