Skip to content

Commit

Permalink
Showing 1 changed file with 6 additions and 0 deletions.
6 changes: 6 additions & 0 deletions google/cloud/aiplatform/constants/prediction.py
Original file line number Diff line number Diff line change
@@ -79,6 +79,12 @@
]

TF_CONTAINER_URIS = [
"us-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-10:latest",
"europe-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-10:latest",
"asia-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-10:latest",
"us-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-10:latest",
"europe-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-10:latest",
"asia-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-10:latest",
"us-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-9:latest",
"europe-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-9:latest",
"asia-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-9:latest",

0 comments on commit 3e95e8d

Please sign in to comment.