diff --git a/.github/workflows/ci-pipeline.yml b/.github/workflows/ci-pipeline.yml
index ac647e2..0413a25 100644
--- a/.github/workflows/ci-pipeline.yml
+++ b/.github/workflows/ci-pipeline.yml
@@ -9,7 +9,7 @@ jobs:
       fail-fast: false
       matrix:
         os: [ubuntu-latest, windows-2022]
-        python-version: ["3.8", "3.9", "3.10", "3.11"]
+        python-version: ["3.8", "3.10", "3.11"]
     steps:
       - uses: actions/checkout@v3
       - if: matrix.os == 'ubuntu-latest'
@@ -21,7 +21,6 @@ jobs:
           python-version: ${{ matrix.python-version }}
       - name: Install requirements
         run: |
-          pip install torch torchvision --index-url https://download.pytorch.org/whl/cpu
           pip install -r test/requirements.txt
           pip install -r test/torch_requirements.txt
           pip install .
@@ -43,7 +42,6 @@ jobs:
           python-version: 3.9
       - name: Generate coverage report
         run: |
-          pip install torch torchvision --index-url https://download.pytorch.org/whl/cpu
           pip install -r test/requirements.txt
           pip install -r test/torch_requirements.txt
           pip install .
@@ -65,8 +63,8 @@ jobs:
           python-version: 3.9
       - name: Install dependencies
         run: |
-          pip install torch torchvision --index-url https://download.pytorch.org/whl/cpu
           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