diff --git a/requirements.txt b/requirements.txt index ad3fd49691d4..931f93646b73 100644 --- 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