From b3e416cd27b764ada141e03964873d27361ec62a Mon Sep 17 00:00:00 2001
From: huchukato <huchukato@gmail.com>
Date: Thu, 5 Mar 2026 07:21:52 +0100
Subject: [PATCH] Improve TensorRT detection with retry mechanism to handle
 concurrent installations

---
 .DS_Store   | Bin 6148 -> 6148 bytes
 __init__.py |  20 +++++++++++++-------
 2 files changed, 13 insertions(+), 7 deletions(-)

diff --git a/.DS_Store b/.DS_Store
index 686ec227da1631c44f531e8c1e4431638aa510e3..ec25b90cef95fa8ec5ba64c2f10707e71e1ae23c 100644
GIT binary patch
delta 106
zcmZoMXffEJ#uVFS!N9=4!l1{H&XCDalAG`1l9ZF51Qg@YQ%pUwdd+c1RQVLV@&y@&
U!O8i#1wcIv415Bco0%F#0igaGod5s;

delta 106
zcmZoMXffEJ#uOWPhk=2Cg+Y%YogtH<Bsbs1B`GIA2`I*4_iPzY*2?3KsPZXz<qI+l
VgOl@f3xIkU7<d~tH#0Sg0sz}F8>;{S

diff --git a/__init__.py b/__init__.py
index dacc93a..94ca94e 100644
--- a/__init__.py
+++ b/__init__.py
@@ -88,13 +88,19 @@ def _auto_install_tensorrt():
         return True
     
     try:
-        # Check if TensorRT is already installed
-        try:
-            import tensorrt
-            print("✅ TensorRT already installed")
-            return True
-        except ImportError:
-            print("🔍 TensorRT not found, detecting CUDA version...")
+        # Check if TensorRT is already installed - with retry mechanism
+        max_retries = 3
+        for attempt in range(max_retries):
+            try:
+                import tensorrt
+                print(f"✅ TensorRT already installed (version: {tensorrt.__version__})")
+                return True
+            except ImportError:
+                if attempt < max_retries - 1:
+                    print(f"🔍 TensorRT not found (attempt {attempt + 1}/{max_retries}), waiting 2 seconds...")
+                    time.sleep(2)  # Wait for potential concurrent installation
+                else:
+                    print("🔍 TensorRT not found after retries, detecting CUDA version...")
         
         # Detect CUDA version
         cuda_version = None