Add TensorRT dependencies (ultralytics#8553)

Update requirements.txt
ctjanuhowski · Sep 8, 2022 · ec89716 · ec89716
1 parent 5c1f033
commit ec89716
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -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