diff --git a/google/cloud/aiplatform/constants/prediction.py b/google/cloud/aiplatform/constants/prediction.py
index e2c039d8aa..a5467fb910 100644
--- a/google/cloud/aiplatform/constants/prediction.py
+++ b/google/cloud/aiplatform/constants/prediction.py
@@ -166,6 +166,12 @@
 ]
 
 PYTORCH_CONTAINER_URIS = [
+    "us-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.2-0:latest",
+    "europe-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.2-0:latest",
+    "asia-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.2-0:latest",
+    "us-docker.pkg.dev/vertex-ai/prediction/pytorch-gpu.2-0:latest",
+    "europe-docker.pkg.dev/vertex-ai/prediction/pytorch-gpu.2-0:latest",
+    "asia-docker.pkg.dev/vertex-ai/prediction/pytorch-gpu.2-0:latest",
     "us-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.1-13:latest",
     "europe-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.1-13:latest",
     "asia-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.1-13:latest",