From 88d8448ed720e58105afc201dc5f7b737b649828 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 35971c3ab107..7fa3e5921bed 100644
--- a/export.py
+++ b/export.py
@@ -336,8 +336,7 @@ def export_coreml(model, im, file, num_boxes, num_classes, labels, conf_thres, i
 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