add installation to toml for xpu

openvinotoolkit · Nov 13, 2024 · 53e5f42 · 53e5f42
1 parent 2fb1a1e
commit 53e5f42
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -72,9 +72,9 @@ docs = [
 ]
 
 xpu = [
-    "torch==2.1.0.post2",
-    "torchvision==0.16.0.post2",
-    "intel-extension-for-pytorch==2.1.30+xpu",
+    "torch==2.6.0.dev20241104+xpu",
+    "torchvision==0.20.0.dev20241104+xpu",
+    "pytorch-triton-xpu==3.1.0+91b14bf559"
     "oneccl_bind_pt==2.1.300+xpu",
     "lightning==2.2",
     "pytorchcv==0.0.67",
@@ -89,7 +89,7 @@ xpu = [
 ]
 
 base = [
-    "torch==2.5.0",
+    "torch==2.5.1",
     "lightning==2.4.0",
     "pytorchcv==0.0.67",
     "timm==1.0.3",