diff --git a/ci/test.py b/ci/test.py
index 7772e4321d4974e699280ae419924b7874d2589d..ab41980d6b6bae18e8bf990f162b20b57edd82af 100755
--- a/ci/test.py
+++ b/ci/test.py
@@ -37,7 +37,6 @@ if __name__ == "__main__":
         ]
     )
 
-    check_call([python_venv_executable(), "-m", "pip", "install", "torch"])
     check_call(
         [
             python_venv_executable(),
diff --git a/setup.py b/setup.py
index 19fae52e44b1e25b23f29a044016b79eafa457cd..14d44d9b7e322e78d8823e2266e77b06d8e2c12e 100644
--- a/setup.py
+++ b/setup.py
@@ -59,6 +59,7 @@ setup(
         "deprecation",
         "tqdm",
         "pre-commit",
+        "torch",
     ],
     classifiers=[
         "Development Status :: 3 - Alpha",