曾逸夫(Zeng Yifu) commited on
Commit
fbd3020
·
unverified ·
1 Parent(s): 0414637

Add TensorRT dependencies (#8553)

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -0
requirements.txt CHANGED
@@ -26,6 +26,8 @@ seaborn>=0.11.0
26
  # coremltools>=4.1 # CoreML export
27
  # onnx>=1.9.0 # ONNX export
28
  # onnx-simplifier>=0.3.6 # ONNX simplifier
 
 
29
  # scikit-learn==0.19.2 # CoreML quantization
30
  # tensorflow>=2.4.1 # TFLite export
31
  # tensorflowjs>=3.9.0 # TF.js export
 
26
  # coremltools>=4.1 # CoreML export
27
  # onnx>=1.9.0 # ONNX export
28
  # onnx-simplifier>=0.3.6 # ONNX simplifier
29
+ # nvidia-pyindex # TensorRT export
30
+ # nvidia-tensorrt # TensorRT export
31
  # scikit-learn==0.19.2 # CoreML quantization
32
  # tensorflow>=2.4.1 # TFLite export
33
  # tensorflowjs>=3.9.0 # TF.js export