16337
bf5ddb3e7a
- requirements.txt: GPU 依赖从注释改为正式声明，统一版本 PyTorch 2.1.2+cu121, TensorRT 8.6.1.6, ultralytics 8.3.5 NumPy 1.24→1.26.4, OpenCV 4.8.0.74→76, 新增 onnx/Pillow 等 - Dockerfile: 基于 nvcr.io/nvidia/tensorrt:23.08-py3 (CUDA 12.1 + cuDNN 8.9 + TRT 8.6) - docker-compose.yml: GPU 访问、host 网络、卷挂载、日志限制 - .dockerignore: 排除模型/数据/日志等大文件 Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
79 lines
2.0 KiB
Plaintext
79 lines
2.0 KiB
Plaintext
# 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
|