Question

要通过 gcloud ai平台（例如gcloud ml-engine）提交培训工作，请使用gcloud SDK中的following command：

gcloud ai-platform jobs submit COMMAND [GCLOUD_WIDE_FLAG …]

我想以编程方式执行此操作，即从python代码（或其他任何语言）执行。

import gcloud-ai-platform as gap
gap.submit_job(COMMAND)

有这样的命令吗？如果不存在，该如何解决？（以编程方式使用gcloud sdk）

Answer 1

我发现有关自定义图片的文档有些混乱。技巧是通过masterConfig / imageUri指定--master-image-uri：

training_inputs = {
    'scaleTier': 'BASIC',
    'packageUris': [
    ],
    'masterConfig': {
        'imageUri': settings["AI_SERVER_URI"]
    },
    'args': [
        "java", "-cp", "MY.jar:jars/*", "io.manycore.Test",
        "jar positional argument"
    ],
    'region': 'us-central1',
    'pythonVersion': '3.7',
    'scheduling': {
        'maxRunningTime': '3600s'
    },
}

job_spec = {'jobId': jobid, 'trainingInput': training_inputs}

project_name = settings["PROJECT_ID"]
project_id = 'projects/{}'.format(project_name)

cloudml = discovery.build('ml', 'v1', credentials=self.credentials)

request = cloudml.projects().jobs().create(body=job_spec, parent=project_id)

Answer 2

要提交培训工作，here有一个可以遵循的示例。

它同时具有gcloud和等效的python代码两种方法。

以编程方式（通过python代码）提交gcloud AI平台培训工作

2 个答案: