Fix dependencies in setup.py

aramis-lab · Aug 25, 2021 · 4b9508e · 4b9508e
1 parent 3fff1f7
commit 4b9508e
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -36,6 +36,14 @@
         "Intended Audience :: Developers",
         "Programming Language :: Python",
     ],
-    install_requires=["numpy>=1.17", "clinica>=0.3.8", "torch", "tensorboard"],
-    python_requires=">=3.6",
+    install_requires=[
+        "numpy>=1.17",
+        "clinica==0.4.1",
+        "torch>=1.8",
+        "tensorboard",
+        "toml",
+        "click>=7.0",
+        "pynvml",
+    ],
+    python_requires=">=3.7",
 )