tensorflow - 如何为 KubeFlow 上的 TFjob 指定/增加 CPU 使用率？

Question

我有一个使用 Kustomize 在最新版本上运行 KubeFlow 的 GKE 设置。主 TFJob 拉取完整模型的 Docker 映像并运行它。我遇到了一个简单的问题，我希望增加 CPU 使用量，但似乎无法做到。

这是我需要在 Docker 模型代码、Kubernetes 配置、TFJob yamls 或其他地方指定的内容吗？

我尝试为容器的 cpu 资源指定最小大小和限制。我还尝试为主 pod 指定这些，但它似乎没有更新。

对此也没有太多文档或许多类似的问题。

标签： tensorflowkubernetesgoogle-cloud-platformgoogle-kubernetes-enginekubeflow

Answer 1

只需参考自定义 TFJob

是的，它是在您的 TFJob yaml 中设置的。代码片段：

resources:
          limits:
            cpu: '1'