如何在 AWS 集群上运行 TensorFlow？答案

【问题标题】：How to run TensorFlow on an AWS cluster?如何在 AWS 集群上运行 TensorFlow？
【发布时间】：2021-05-15 20:41:52
【问题描述】：

我正在尝试在 EMR/EC2 集群上运行分布式张量流，但我不知道如何在集群中指定不同的实例来运行部分代码。

在文档中，他们使用tf.device("/gpu:0") 来指定 GPU。但是，如果我有一个主 CPU 和 5 个不同的从属 GPU 实例在 EMR 集群中运行，并且我想指定这些 GPU 来运行一些代码怎么办？我无法使用实例的公共 DNS 名称输入 tf.device()，因为它会引发错误，指出无法解析名称。

【问题讨论】：

标签： python amazon-web-services amazon-ec2 tensorflow

【解决方案1】：

自从您提出问题以来，AWS 已经发布了一些代码来简化在 EC2 集群上使用分布式 TensorFlow。

看到这个github repository。 README.md 中描述了所有内容，但简短的故事是它将创建一个 AWS 堆栈与

安全组
弹性文件系统
安装了 AWS 深度学习 AMI 和 EFS 的 EC2 实例，
将配置 EC2 实例，以便您可以通过在主节点上运行命令轻松运行分布式 TensorFlow 运行（请参阅在 TensorFlow 上运行分布式训练部分）。

【讨论】：