diff --git a/.github/workflows/ci-pipeline.yml b/.github/workflows/ci-pipeline.yml
index 3922aac..0413a25 100644
--- 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
index 4cc375e..67a09d8 100644
--- 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
index 729804f..fd14302 100644
--- 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