missed onnxruntime-gpu

neuralmagic · Nov 30, 2023 · d27e941 · d27e941
1 parent 73adbc9
commit d27e941
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -39,7 +39,7 @@
     "tensorboard>=2.0.0",
     "setuptools>=56.0.0",
     "optuna>=3.0.2",
-    "onnxruntime",
+    "onnxruntime-gpu",
 ]
 
 _nm_deps = [