From d3e5fe8a3f90184b19cb9178b45ce34265887efb Mon Sep 17 00:00:00 2001 From: Karl Bartel Date: Fri, 19 Nov 2021 08:24:00 +0100 Subject: [PATCH] Revert "Use CPU-only torch package" This reverts commit a280a73d32c36371f2d78ae7bfaac5e9365a9243. --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index 4b3cd06..eca948d 100644 --- 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",