add sparsifyml dependencies to sparsify install (#195)

neuralmagic · May 1, 2023 · afc5ece · afc5ece
1 parent d5d9b48
commit afc5ece
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -31,10 +31,19 @@
 _PACKAGE_NAME = "sparsify" if is_release else "sparsify-nightly"
 
 
-_deps = ["pydantic>=1.8.2", "pyyaml>=5.0.0", "click~=8.0.0", "tensorboard>=2.0.0"]
+_deps = [
+    "pydantic>=1.8.2",
+    "pyyaml>=5.0.0",
+    "click~=8.0.0",
+    "tensorboard>=2.0.0",
+    "setuptools>=56.0.0",
+    "optuna>=3.0.2",
+    "onnxruntime-gpu",
+]
 _nm_deps = [
     f"{'sparsezoo' if is_release else 'sparsezoo-nightly'}~={version_nm_deps}",
     f"{'sparseml' if is_release else 'sparseml-nightly'}[torchvision]~={version_nm_deps}",  # noqa E501
+    f"{'deepsparse' if is_release else 'deepsparse-nightly'}~={version_nm_deps}",
 ]