TensorFlow 是一个开源的机器学习框架,可以在多种平台上运行,包括 Linux。以下是在 Linux 系统上安装 TensorFlow 的指南。

安装步骤

  1. 更新系统包
    确保您的系统包是最新的:

    sudo apt-get update
    sudo apt-get upgrade
    
  2. 安装依赖项
    TensorFlow 需要一些依赖项。以下命令将安装所有必需的依赖项:

    sudo apt-get install -y python3 python3-pip python3-dev
    
  3. 创建虚拟环境
    创建一个虚拟环境以避免安装 TensorFlow 时影响到系统 Python:

    python3 -m venv tensorflow_env
    source tensorflow_env/bin/activate
    
  4. 安装 TensorFlow
    使用 pip 安装 TensorFlow:

    pip install tensorflow
    
  5. 验证安装
    运行以下 Python 代码来验证 TensorFlow 是否已正确安装:

    import tensorflow as tf
    print(tf.__version__)
    

注意事项

  • 确保您的 Python 版本与 TensorFlow 兼容。
  • 如果您需要 GPU 支持的 TensorFlow,请使用 tensorflow-gpu 而不是 tensorflow

扩展阅读

更多关于 TensorFlow 的信息,请访问我们的 TensorFlow 教程页面

TensorFlow Logo