以下是 TensorFlow 在不同操作系统上的安装步骤。

Windows 安装

  1. 打开 PowerShell 或命令提示符。

  2. 运行以下命令安装 Python 3.6 或更高版本:

    python -m pip install --upgrade pip
    
  3. 安装 TensorFlow:

    pip install tensorflow
    
  4. 安装 TensorFlow GPU 支持(如果需要):

    pip install tensorflow-gpu
    

macOS 安装

  1. 打开终端。

  2. 运行以下命令安装 Homebrew:

    /bin/bash -c "$(curl -fsSL https://raw.githubusercontent.com/Homebrew/install/HEAD/install.sh)"
    
  3. 安装 Python 3:

    brew install python
    
  4. 安装 TensorFlow:

    pip install tensorflow
    
  5. 安装 TensorFlow GPU 支持(如果需要):

    pip install tensorflow-gpu
    

Linux 安装

  1. 打开终端。

  2. 安装 Python 3:

    sudo apt-get update
    sudo apt-get install python3 python3-pip
    
  3. 安装 TensorFlow:

    pip3 install tensorflow
    
  4. 安装 TensorFlow GPU 支持(如果需要):

    pip3 install tensorflow-gpu
    

验证安装

安装完成后,可以通过以下命令验证 TensorFlow 是否安装成功:

python -c "import tensorflow as tf; print(tf.reduce_sum(tf.random.normal([1000, 1000])))"

如果一切正常,您将看到类似以下输出:

0.9988336

更多关于 TensorFlow 的信息,请访问我们的官方文档

相关资源

TensorFlow Logo