TensorFlow 2 - 深度学习 AMI
AWS 文档中描述的 AWS 服务或功能可能因区域而异。要查看适用于中国区域的差异,请参阅中国的 AWS 服务入门

本文属于机器翻译版本。若本译文内容与英语原文存在差异,则一律以英文原文为准。

TensorFlow 2

本教程介绍如何在运行使用 Conda (DLAAMI) 的实例上激活 TensorFlow 2 并运行 TensorFlow 2 程序。

当框架的稳定的 Conda 程序包发布时,它会在 DLAMI 上进行测试并预安装。如果您希望运行最新的、未经测试的每日构建版本,您可以手动安装 TensorFlow 2 的每日构建版本(试验)

激活 TensorFlow 2

在 DLAMI 上运行 TensorFlow

  1. 要激活 TensorFlow 2,请使用 Conda 打开 DLAMI 的 Amazon Elastic Compute Cloud (Amazon EC2) 实例。

  2. 对于具有 CUDA 10.1 和 MKL-DNN 的 Python 3 上的 TensorFlow 2 和 Keras 2,运行以下命令:

    $ source activate tensorflow2_p36
  3. 启动 iPython 终端:

    (tensorflow2_p36)$ ipython
  4. 运行 TensorFlow 2 程序以验证其是否正常运行:

    import tensorflow as tf hello = tf.constant('Hello, TensorFlow!') tf.print(hello)

    Hello, TensorFlow! 应显示在您的屏幕上。

安装 TensorFlow 2 的每日构建版本(试验)

您可以将最新的 TensorFlow 2 版本安装到您的深度学习 AMI 上的任一或两个 TensorFlow 2 Conda 环境。

从每日构建版本安装 TensorFlow

  1. 对于 Python 3 TensorFlow 2 环境,运行以下命令:

    $ source activate tensorflow2_p36
  2. 删除当前安装的 TensorFlow。

    注意

    其余步骤假定您使用的是 tensorflow2_p36 环境。

    (tensorflow2_p36)$ pip uninstall tensorflow
  3. 安装 TensorFlow 的最新每日构建版本。

    (tensorflow2_p36)$ pip install tf-nightly
  4. 要验证您是否已成功安装最新的每日构建版本,请启动 IPython 终端并检查 TensorFlow 版本。

    (tensorflow2_p36)$ ipython
    import tensorflow print (tensorflow.__version__)

    输出应类似于以下内容:2.1.0-dev20191122

更多教程

有关教程,请参阅名为的文件夹Deep Learning AMI with Conda tutorials在 DLAMI 的主目录中。

有关更多教程和示例,请参阅TensorFlow Python API或者请参阅TensorFlow网站.