Skip to content

Commit f7b5b54

Browse files
authoredOct 17, 2024··
Install torchvision CPU in OpenVINO notebook tests (#953)
1 parent 981442f commit f7b5b54

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed
 

‎.github/workflows/test_openvino_notebooks.yml

+1-1
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,7 @@ jobs:
3939
# Install PyTorch CPU to prevent unnecessary downloading/installing of CUDA packages
4040
# ffmpeg, torchaudio and pillow are required for image classification and audio classification pipelines
4141
sudo apt-get install ffmpeg
42-
pip install torch torchaudio --extra-index-url https://download.pytorch.org/whl/cpu
42+
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
4343
pip install -r notebooks/openvino/requirements.txt
4444
pip install .[tests,openvino] nbval
4545

0 commit comments

Comments
 (0)
Please sign in to comment.