Add TensorRT dependencies (#8553)

Update requirements.txt
This commit is contained in:
曾逸夫(Zeng Yifu) authored and GitHub committed 2022-07-12 13:19:25 +02:00
1 parent 04146371b9
commit fbd3020525
1 file changed
+2
+2
View File
@@ -26,6 +26,8 @@ seaborn>=0.11.0
# coremltools>=4.1 # CoreML export
# onnx>=1.9.0 # ONNX export
# onnx-simplifier>=0.3.6 # ONNX simplifier
# nvidia-pyindex # TensorRT export
# nvidia-tensorrt # TensorRT export
# scikit-learn==0.19.2 # CoreML quantization
# tensorflow>=2.4.1 # TFLite export
# tensorflowjs>=3.9.0 # TF.js export