✨ feat(deploy): Modernize TRT deployment for Blackwell with docs (#620)

* ✨ feat(deploy): Modernize TRT deployment for Blackwell with docs

* 📝 Update README.md

Author: Hitbee-dev

rtdetrv2_pytorch/Dockerfile

@@ -1,5 +1,4 @@
-# tensorrt:23.01-py3 (8.5.2.2)
-FROM nvcr.io/nvidia/tensorrt:23.01-py3
+FROM nvcr.io/nvidia/pytorch:25.06-py3
 
 WORKDIR /workspace
 
@@ -8,4 +7,4 @@ COPY requirements.txt .
 RUN pip install --upgrade pip && \
     pip install -r requirements.txt
 
-CMD ["/bin/bash"]
+CMD ["/bin/bash"]

rtdetrv2_pytorch/docker-compose.yml

@@ -1,15 +1,23 @@
-version: "3.9"
-
 services:
   tensorrt-container:
     build:
       context: .
       dockerfile: Dockerfile
-    image: rtdetr-v2:23.01
+    image: rtdetr-v2:25.06
+    container_name: rtdetr-v2-trt
+    ports:
+      - "6006:6006" # tensorboard
     volumes:
       - ./:/workspace
-    runtime: nvidia
-    environment:
-      - NVIDIA_VISIBLE_DEVICES=all
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              count: all
+              capabilities: [gpu]
+    working_dir: /workspace
+    restart: unless-stopped
     stdin_open: true
     tty: true
+    command: bash