曾逸夫(Zeng Yifu)
commited on
Add TensorRT dependencies (#8553)
Browse files- 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
|