natas_hw
/
openvino
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Use CPU torch (#24658) 

### Details:
- See
https://github.com/openvinotoolkit/openvino/actions/runs/9195359216/job/25292054679#step:16:224
This commit is contained in:
Ilya Lavrenov 2024-05-25 22:10:38 +04:00 committed by GitHub
parent 0fe524bcbe
commit 72028b7bbc
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/bindings/python/src/openvino/preprocess/torchvision/requirements.txt
View File

@ -1,3 +1,4 @@
--extra-index-url https://download.pytorch.org/whl/cpu
torch>=1.13 torch>=1.13
torchvision; platform_machine == 'arm64' and python_version >= '3.8' torchvision; platform_machine == 'arm64' and python_version >= '3.8'
torchvision; platform_machine != 'arm64' torchvision; platform_machine != 'arm64'

6
tests/constraints.txt
View File

@ -15,7 +15,7 @@ fastjsonschema~=2.17.1
tensorflow>=2.5,<2.17.0 tensorflow>=2.5,<2.17.0
requests>=2.25.1 requests>=2.25.1
opencv-python>=4.5 opencv-python>=4.5
paddlepaddle==2.5.0 paddlepaddle==2.6.0
protobuf>=3.18.1,<4.0.0 protobuf>=3.18.1,<4.0.0
py>=1.9.0 py>=1.9.0
pytest>=5.0,<7.5 pytest>=5.0,<7.5
@ -24,7 +24,9 @@ pytest-html==4.1.1
pytest-timeout==2.2.0 pytest-timeout==2.2.0
jax<=0.4.14 jax<=0.4.14
jaxlib<=0.4.14 jaxlib<=0.4.14
torch>=1.13,<2.3
kornia==0.7.0 kornia==0.7.0
networkx<=3.1.0 networkx<=3.1.0
keras>=2.0.0,<3.0.0 keras>=2.0.0,<3.0.0
--extra-index-url https://download.pytorch.org/whl/cpu
torch>=1.13,<2.3

1
tests/model_hub_tests/pytorch/requirements.txt
View File

@ -1,5 +1,4 @@
-c ../../constraints.txt -c ../../constraints.txt
--extra-index-url https://download.pytorch.org/whl/cpu
auto-gptq>=0.5.1 auto-gptq>=0.5.1
av av
basicsr basicsr