Skip to content

Commit 2e35263

Browse files
vertex-sdk-botcopybara-github
authored andcommitted
chore: Add PyTorch 1.12 images to prediction first-party uris.
PiperOrigin-RevId: 504886881
1 parent d29195b commit 2e35263

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

google/cloud/aiplatform/constants/prediction.py

+6
Original file line numberDiff line numberDiff line change
@@ -148,6 +148,12 @@
148148
]
149149

150150
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",
151157
"us-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.1-11:latest",
152158
"europe-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.1-11:latest",
153159
"asia-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.1-11:latest",

0 commit comments

Comments
 (0)