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",