From f7b5b547c167cb6a9f20fa77d493ee2dde3c3034 Mon Sep 17 00:00:00 2001 From: Helena Kloosterman Date: Thu, 17 Oct 2024 10:17:22 +0200 Subject: [PATCH] Install torchvision CPU in OpenVINO notebook tests (#953) --- .github/workflows/test_openvino_notebooks.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/test_openvino_notebooks.yml b/.github/workflows/test_openvino_notebooks.yml index ded091d5ad..26a09012ff 100644 --- a/.github/workflows/test_openvino_notebooks.yml +++ b/.github/workflows/test_openvino_notebooks.yml @@ -39,7 +39,7 @@ jobs: # Install PyTorch CPU to prevent unnecessary downloading/installing of CUDA packages # ffmpeg, torchaudio and pillow are required for image classification and audio classification pipelines sudo apt-get install ffmpeg - pip install torch torchaudio --extra-index-url https://download.pytorch.org/whl/cpu + pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu pip install -r notebooks/openvino/requirements.txt pip install .[tests,openvino] nbval