diff --git a/.github/workflows/testing.yml b/.github/workflows/testing.yml
index 9cf570a..9d14354 100644
--- a/.github/workflows/testing.yml
+++ b/.github/workflows/testing.yml
@@ -48,8 +48,9 @@ jobs:
       - name: Install package
         run: |
           python -m pip install --upgrade pip
-          pip install torch
-          pip install -e .[test]
+          pip install torch==2.1.1
+          pip install torch-scatter -f https://data.pyg.org/whl/torch-2.1.0+cpu.html
+          pip install -e .[strict,test]
 
       - name: Test
         run: pytest
diff --git a/pyproject.toml b/pyproject.toml
index 99048c8..77ec847 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -8,6 +8,7 @@ version = "0.0.1"
 readme = "README.md"
 requires-python = ">=3.8, <3.11"
 dependencies = [
+    "torch>=2.0.0",
     "pytorch-lightning>=2.0.0",
     "torchmetrics>=0.10.0, <1.0.0",
     "torch_geometric>=2.3.0",