Revert "Use CPU-only torch package"

This reverts commit a280a73.
realrate · Nov 19, 2021 · d3e5fe8 · d3e5fe8
1 parent 5fd3d01
commit d3e5fe8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -25,7 +25,7 @@
         "pandas==1.1.5",
         "scipy==1.5.4",
         "sympy==1.5.1",
-        "torch@https://download.pytorch.org/whl/cpu/torch-1.9.1%2Bcpu-cp39-cp39-linux_x86_64.whl",
+        "torch==1.9.1",
         "pre-commit",
     ],
     python_requires=">=3.9",