More strict dependencies for PyTorch in torchvision preprocess

ababushk · Jan 28, 2025 · 946eb0f · 946eb0f
1 parent 45622c1
commit 946eb0f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/bindings/python/src/openvino/preprocess/torchvision/requirements.txt b/src/bindings/python/src/openvino/preprocess/torchvision/requirements.txt
@@ -1,5 +1,6 @@
 --extra-index-url https://download.pytorch.org/whl/cpu
-torch>=1.13
+torch<=2.5.1; python_version <= '3.9'
+torch>=1.13; python_version > '3.9'
 torchvision; platform_machine == 'arm64' and python_version >= '3.9'
 torchvision; platform_machine != 'arm64'
 pillow>=9.0