We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent d29195b commit 2e35263Copy full SHA for 2e35263
google/cloud/aiplatform/constants/prediction.py
@@ -148,6 +148,12 @@
148
]
149
150
PYTORCH_CONTAINER_URIS = [
151
+ "us-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.1-12:latest",
152
+ "europe-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.1-12:latest",
153
+ "asia-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.1-12:latest",
154
+ "us-docker.pkg.dev/vertex-ai/prediction/pytorch-gpu.1-12:latest",
155
+ "europe-docker.pkg.dev/vertex-ai/prediction/pytorch-gpu.1-12:latest",
156
+ "asia-docker.pkg.dev/vertex-ai/prediction/pytorch-gpu.1-12:latest",
157
"us-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.1-11:latest",
158
"europe-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.1-11:latest",
159
"asia-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.1-11:latest",
0 commit comments