pin torchvision version to something compatible with torch 2.1

neuromorphs · May 15, 2024 · 2fb1664 · 2fb1664
1 parent b81f886
commit 2fb1664
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 3 deletions.
diff --git a/.github/workflows/ci-pipeline.yml b/.github/workflows/ci-pipeline.yml
@@ -63,8 +63,8 @@ jobs:
  python-version: 3.9
  - name: Install dependencies
  run: |
- pip install -r test/torch_requirements.txt
  pip install -r docs/requirements.txt
+ pip install -r test/torch_requirements.txt
  pip install .
  - name: Build documentation
  run: cd docs && make clean && make html # Use SPHINXOPTS="-W" to fail on warning.

diff --git a/docs/requirements.txt b/docs/requirements.txt
@@ -4,8 +4,6 @@ sphinx-book-theme
 sphinx-gallery
 myst_nb
 pbr
-torchvision
 ipywidgets
 matplotlib
-torchdata
 sphinx-autoapi
diff --git a/test/torch_requirements.txt b/test/torch_requirements.txt
@@ -1,4 +1,5 @@
 --index-url https://download.pytorch.org/whl/cpu
 torch==2.1.0
 torchaudio==2.1.0
+torchvision==0.16.0
 torchdata