From cdd557e5e86b0b4d4cf401509aba5914e0bab8b7 Mon Sep 17 00:00:00 2001 From: Jaycee Li Date: Tue, 29 Nov 2022 13:48:39 -0800 Subject: [PATCH] fix: Add pre-built container(tf2-gpu-2-1) to the container URI list PiperOrigin-RevId: 491727625 --- google/cloud/aiplatform/constants/prediction.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/google/cloud/aiplatform/constants/prediction.py b/google/cloud/aiplatform/constants/prediction.py index 88581c8e44..52a69ab692 100644 --- a/google/cloud/aiplatform/constants/prediction.py +++ b/google/cloud/aiplatform/constants/prediction.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2022 Google LLC # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. @@ -136,6 +136,9 @@ "us-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-1:latest", "europe-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-1:latest", "asia-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-1:latest", + "us-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-1:latest", + "europe-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-1:latest", + "asia-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-1:latest", "us-docker.pkg.dev/vertex-ai/prediction/tf-cpu.1-15:latest", "europe-docker.pkg.dev/vertex-ai/prediction/tf-cpu.1-15:latest", "asia-docker.pkg.dev/vertex-ai/prediction/tf-cpu.1-15:latest",