security-ai-edge/requirements.txt

# Edge_Inference_Service 依赖清单
# 安装命令: pip install -r requirements.txt
# 环境要求: Python 3.10 | CUDA 12.1 | cuDNN 8.9 | TensorRT 8.6.1
# Docker 基础镜像: nvcr.io/nvidia/tensorrt:23.08-py3

# ============================================================
# GPU 推理依赖（TensorRT 8.6 + CUDA 12.1）
# ============================================================

# PyTorch - CUDA 12.1 下最稳定版本
# 安装命令: pip install torch==2.1.2 torchvision==0.16.2 --index-url https://download.pytorch.org/whl/cu121
--extra-index-url https://download.pytorch.org/whl/cu121
torch==2.1.2
torchvision==0.16.2

# TensorRT Python 绑定（NGC 镜像已预装，裸机需手动安装）
tensorrt==8.6.1.6
pycuda==2023.1.1

# YOLO11 目标检测框架
ultralytics==8.3.5

# ONNX 模型转换与优化
onnx==1.16.0
onnxsim==0.4.36
onnxruntime-gpu==1.17.1

# ============================================================
# 核心依赖
# ============================================================

# 视频处理
opencv-python==4.8.0.76

# 数值计算（锁定 1.x，避开 NumPy 2.0 破坏性变更）
numpy==1.26.4

# 图像处理
Pillow==10.2.0

# ============================================================
# 数据库依赖
# ============================================================

# ORM 框架
sqlalchemy==2.0.23

# MySQL 驱动
pymysql==1.1.0

# ============================================================
# 消息队列与缓存
# ============================================================

# MQTT 客户端（1.x 最终稳定版）
paho-mqtt==1.6.1

# Redis 客户端
redis==4.6.0

# 腾讯云 COS SDK（截图上传）
cos-python-sdk-v5>=1.9.30

# ============================================================
# 工具库
# ============================================================

pyyaml==6.0.1
requests==2.31.0
psutil==5.9.8
python-dotenv==1.0.1

# ============================================================
# 测试依赖
# ============================================================

pytest==7.4.4
pytest-cov==4.1.0