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

安装 TensorFlow

  1. 安装 Python 环境:确保您的系统中已经安装了 Python 3。可以使用以下命令检查 Python 版本:

    python3 --version
    

    如果没有安装 Python 3,可以从 Python 官网 下载并安装。

  2. 创建虚拟环境:为了保持项目的依赖项清晰,建议创建一个虚拟环境。使用以下命令创建一个名为 tensorflow_env 的虚拟环境:

    python3 -m venv tensorflow_env
    

    进入虚拟环境:

    source tensorflow_env/bin/activate
    
  3. 安装 TensorFlow:在虚拟环境中,使用以下命令安装 TensorFlow:

    pip install tensorflow
    

    如果您需要 GPU 支持,可以使用以下命令安装 TensorFlow GPU 版本:

    pip install tensorflow-gpu
    
  4. 验证安装:安装完成后,可以运行以下 Python 代码来验证 TensorFlow 是否已正确安装:

    import tensorflow as tf
    print(tf.__version__)
    

    如果输出 TensorFlow 的版本号,则表示安装成功。

获取更多帮助

如果您在安装过程中遇到任何问题,可以访问 TensorFlow 官方文档 获取更多帮助。

图片展示

TensorFlow 的标志

TensorFlow