diff --git a/packages/syft/setup.cfg b/packages/syft/setup.cfg
index 009c7a3a47e..e3a9774ca8e 100644
--- a/packages/syft/setup.cfg
+++ b/packages/syft/setup.cfg
@@ -62,7 +62,7 @@ syft =
     safetensors==0.4.0
     transformers==4.34.0
     evaluate==0.4.1
-    torch==2.1.0
+    torch[cpu]==2.1.0
     recordlinkage==0.16
     argon2-cffi==23.1.0
 
diff --git a/tox.ini b/tox.ini
index 2a8ceeaa162..eead6745fe8 100644
--- a/tox.ini
+++ b/tox.ini
@@ -26,13 +26,12 @@ envlist =
     syftcli.build
 skipsdist = True
 
+
 [testenv]
 basepython = python3
-install_command = pip install --find-links https://download.pytorch.org/whl/cpu/torch_stable.html {opts} {packages}
+install_command = pip install {opts} {packages}
 commands =
     python --version
-setenv =
-    PIP_FIND_LINKS=https://download.pytorch.org/whl/cpu/torch_stable.html
 
 # Syft
 [testenv:syft]