| 123456789101112131415161718192021222324252627282930313233 |
- # LightGlue TensorRT 优化版本依赖清单
- # GTX 1660 专用
- # ============================================
- # 基础依赖（必需）
- # ============================================
- numpy>=1.19.0
- opencv-python>=4.5.0
- matplotlib>=3.3.0
- kornia>=0.6.11
- # ============================================
- # PyTorch (CUDA 12.1)
- # ============================================
- # 注意：这些包需要从 PyTorch 官方源安装
- # 安装命令：
- # pip install torch==2.5.1 torchvision==0.20.1 torchaudio==2.5.1 --index-url https://download.pytorch.org/whl/cu121
- torch==2.5.1
- torchvision==0.20.1
- torchaudio==2.5.1
- # ============================================
- # TensorRT 优化（二选一）
- # ============================================
- # 方案一：torch-tensorrt（推荐，最简单）
- torch-tensorrt==2.5.0
- # 方案二：ONNX → TensorRT（如果方案一失败，使用此方案）
- # nvidia-tensorrt>=8.0.0
- # onnx>=1.12.0
- # onnxruntime-gpu>=1.12.0
|
