Update setup.py

2026-09-09 11:32:07 +08:00 · 2019-06-06 18:04:58 -04:00
parent 15ca7a40a6
commit eb5b3cfee1
1 changed files with 1 additions and 1 deletions
@@ -17,7 +17,7 @@ setup(
    keywords=["deep learning", "pytorch", "AI"],
    python_requires=">=3.5",
    install_requires=[
-        "torch",
+        "torch>=1.0.0",
        "tqdm",
        "test-tube",
    ],