diff --git a/pyproject.toml b/pyproject.toml
index bf98ec0..571780a 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -47,15 +47,15 @@ Repository = "https://github.com/libxsmm/AI-bench"
 
 [project.optional-dependencies]
 cpu = [
-  "torch==2.9.1+cpu",
+  "torch==2.12.0",
 ]
 xpu = [
-  "torch==2.9.1+xpu",
+  "torch==2.12.0",
   "pytorch-triton-xpu",
   "helion==1.0.0",
 ]
 cuda = [
-  "torch==2.9.1+cu128",
+  "torch==2.12.0",
   "torchvision==0.24.1",
   "pytorch-triton",
   "helion==1.0.0",