fix: Add pre-built container(tf2-gpu-2-1) to the container URI list

jaycee-li · copybara-github · commit cdd557e5e86b · 2022-11-29T13:50:08.000-08:00
PiperOrigin-RevId: 491727625
diff --git 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",
