Add pycuda dependency to setup.py (#49)

onnx · Sep 11, 2018 · f31e57d · f31e57d
1 parent 27cec99
commit f31e57d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -89,4 +89,5 @@
           "numpy>=1.8.1",
           "tensorrt>=3.0.0",
           "onnx>=1.0.1",
+          "pycuda",
       ])