From 72028b7bbc5bf358571766d15f34e222e0278192 Mon Sep 17 00:00:00 2001 From: Ilya Lavrenov Date: Sat, 25 May 2024 22:10:38 +0400 Subject: [PATCH] Use CPU torch (#24658) ### Details: - See https://github.com/openvinotoolkit/openvino/actions/runs/9195359216/job/25292054679#step:16:224 --- .../src/openvino/preprocess/torchvision/requirements.txt | 1 + tests/constraints.txt | 6 ++++-- tests/model_hub_tests/pytorch/requirements.txt | 1 - 3 files changed, 5 insertions(+), 3 deletions(-) diff --git a/src/bindings/python/src/openvino/preprocess/torchvision/requirements.txt b/src/bindings/python/src/openvino/preprocess/torchvision/requirements.txt index 055f77b99d6..23ba17d4918 100644 --- a/src/bindings/python/src/openvino/preprocess/torchvision/requirements.txt +++ b/src/bindings/python/src/openvino/preprocess/torchvision/requirements.txt @@ -1,3 +1,4 @@ +--extra-index-url https://download.pytorch.org/whl/cpu torch>=1.13 torchvision; platform_machine == 'arm64' and python_version >= '3.8' torchvision; platform_machine != 'arm64' diff --git a/tests/constraints.txt b/tests/constraints.txt index 656ae4408f7..b964d71d915 100644 --- a/tests/constraints.txt +++ b/tests/constraints.txt @@ -15,7 +15,7 @@ fastjsonschema~=2.17.1 tensorflow>=2.5,<2.17.0 requests>=2.25.1 opencv-python>=4.5 -paddlepaddle==2.5.0 +paddlepaddle==2.6.0 protobuf>=3.18.1,<4.0.0 py>=1.9.0 pytest>=5.0,<7.5 @@ -24,7 +24,9 @@ pytest-html==4.1.1 pytest-timeout==2.2.0 jax<=0.4.14 jaxlib<=0.4.14 -torch>=1.13,<2.3 kornia==0.7.0 networkx<=3.1.0 keras>=2.0.0,<3.0.0 + +--extra-index-url https://download.pytorch.org/whl/cpu +torch>=1.13,<2.3 \ No newline at end of file diff --git a/tests/model_hub_tests/pytorch/requirements.txt b/tests/model_hub_tests/pytorch/requirements.txt index b41978a0b2e..882be0f61ec 100644 --- a/tests/model_hub_tests/pytorch/requirements.txt +++ b/tests/model_hub_tests/pytorch/requirements.txt @@ -1,5 +1,4 @@ -c ../../constraints.txt ---extra-index-url https://download.pytorch.org/whl/cpu auto-gptq>=0.5.1 av basicsr