在 GCP 上运行 TFF 模拟

本教程介绍如何在 GCP 上运行 TFF 模拟。

在单个运行时容器上运行模拟

1. 安装并初始化 Cloud SDK。

2. 克隆 TensorFlow Federated 仓库。

$ git clone https://github.com/tensorflow/federated.git
$ cd "federated"

3. 运行单个运行时容器。

构建运行时容器。

$ docker build \
    --network=host \
    --tag "<registry>/tff-runtime" \
    --file "tensorflow_federated/tools/runtime/container/Dockerfile" \
    .

发布运行时容器。
```
$ docker push <registry>/tff-runtime
```
创建 Compute Engine 实例
1. 在 Cloud Console 中，转到 VM Instances 页面。
2. 点击 Create instance。
3. 在 Firewall 部分，选择 Allow HTTP traffic 和 Allow HTTPS traffic。
4. 点击 Create 以创建实例。
使用 ssh 连接到实例。
```
$ gcloud compute ssh <instance>
```

在后台运行运行时容器。

$ docker run \
    --detach \
    --name=tff-runtime \
    --publish=8000:8000 \
    <registry>/tff-runtime

退出实例。
```
$ exit
```

获取实例的内部 IP 地址。

此信息稍后将用作我们测试脚本的参数。

$ gcloud compute instances describe <instance> \
    --format='get(networkInterfaces[0].networkIP)'

4. 在客户端容器上运行模拟。

构建客户端容器。

$ docker build \
    --network=host \
    --tag "<registry>/tff-client" \
    --file "tensorflow_federated/tools/client/latest.Dockerfile" \
    .

发布客户端容器。
```
$ docker push <registry>/tff-client
```
创建 Compute Engine 实例
1. 在 Cloud Console 中，转到 VM Instances 页面。
2. 点击 Create instance。
3. 在 Firewall 部分，选择 Allow HTTP traffic 和 Allow HTTPS traffic。
4. 点击 Create 以创建实例。

将实验复制到 Compute Engine 实例中。

$ gcloud compute scp \
    "tensorflow_federated/tools/client/test.py" \
    <instance>:~

使用 ssh 连接到实例。
```
$ gcloud compute ssh <instance>
```

以交互方式运行客户端容器。

随即会在终端打印出字符串“Hello World”。

$ docker run \
    --interactive \
    --tty \
    --name=tff-client \
    --volume ~/:/simulation \
    --workdir /simulation \
    <registry>/tff-client \
    bash

运行 Python 脚本。

使用运行运行时容器的实例的内部 IP 地址。
```
$ python3 test.py --host '<IP address>'
```
退出容器。
```
$ exit
```
退出实例。
```
$ exit
```