Skip to content

Commit cdd557e

Browse files
jaycee-licopybara-github
authored andcommitted
fix: Add pre-built container(tf2-gpu-2-1) to the container URI list
PiperOrigin-RevId: 491727625
1 parent 8779df5 commit cdd557e

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

google/cloud/aiplatform/constants/prediction.py

+4-1
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
# Copyright 2021 Google LLC
1+
# Copyright 2022 Google LLC
22
#
33
# Licensed under the Apache License, Version 2.0 (the "License");
44
# you may not use this file except in compliance with the License.
@@ -136,6 +136,9 @@
136136
"us-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-1:latest",
137137
"europe-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-1:latest",
138138
"asia-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-1:latest",
139+
"us-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-1:latest",
140+
"europe-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-1:latest",
141+
"asia-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-1:latest",
139142
"us-docker.pkg.dev/vertex-ai/prediction/tf-cpu.1-15:latest",
140143
"europe-docker.pkg.dev/vertex-ai/prediction/tf-cpu.1-15:latest",
141144
"asia-docker.pkg.dev/vertex-ai/prediction/tf-cpu.1-15:latest",

0 commit comments

Comments
 (0)