Merge pull request #8195 from madhavajay/madhava/torch_cpu

Trying cpu extras to speed up pip install on tox tasks
OpenMined · Oct 26, 2023 · 5625eb0 · 5625eb0
2 parents a8d7228 + 51a1ef6
commit 5625eb0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git 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
@@ -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]