From fdbb96cc4f8df47e59e250cdb9d405bf5d39aad5 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Fri, 15 Apr 2022 21:48:52 +0200
Subject: [PATCH] Remove `tensorrt` pip install check (#7439)

---
 export.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/export.py b/export.py
index 00b98517cdf6..f97df4710b6f 100644
--- a/export.py
+++ b/export.py
@@ -209,8 +209,7 @@ def export_coreml(model, im, file, prefix=colorstr('CoreML:')):
 def export_engine(model, im, file, train, half, simplify, workspace=4, verbose=False, prefix=colorstr('TensorRT:')):
     # YOLOv5 TensorRT export https://developer.nvidia.com/tensorrt
     try:
-        check_requirements(('tensorrt',))
-        import tensorrt as trt
+        import tensorrt as trt  # pip install -U nvidia-tensorrt --index-url https://pypi.ngc.nvidia.com
 
         if trt.__version__[0] == '7':  # TensorRT 7 handling https://github.com/ultralytics/yolov5/issues/6012
             grid = model.model[-1].anchor_grid