小编Ahm*_*iuk的帖子

我想通过 image_uri 参数将 ECR 上的公共 Docker 映像传递给 Estimator。然而，训练作业未能找到该图像。我正在使用 AWS SageMaker Notebook 实例运行我的代码。

from sagemaker.estimator import Estimator

byoc_image_uri = 'public.ecr.aws/v8x6t9d9/train-image-classifier-firefly'

estimator = Estimator(image_uri=byoc_image_uri,
                      role=role,
                      base_job_name='tf-custom-container-test-job',
                      instance_count=1,
                      hyperparameters=hyperparameters,
                      output_path=output_path,
                      instance_type='ml.p2.xlarge')

我尝试过的：

• 我首先尝试在 ECR privet 存储库上使用相同的图像。这有效
• 然后将相同的映像推送到我创建的公共 ECR 存储库。图像已成功推送，我可以成功拉取该图像。
• 但是，当我尝试通过 Estimator image_uri 参数使用相同的图像 url 时。我收到以下错误消息

    ClientError: An error occurred (ValidationException) when calling the CreateTrainingJob operation: TrainingImageConfig with TrainingRepositoryAccessMode set to VPC must be provided when using a training image from a private Docker registry. Please provideTrainingImageConfig and TrainingRepositoryAccessMode set to VPC when using …