LightGlueProject/LightGlue_Deployment/fix_torch_dependencies.bat

@echo off
REM 修复 PyTorch 版本冲突
REM 确保 torch, torchvision, torchaudio 版本匹配

echo ========================================
echo 修复 PyTorch 版本依赖冲突
echo ========================================
echo.

echo 步骤 1: 卸载冲突的包...
pip uninstall -y torch torchvision torchaudio torch-tensorrt

echo.
echo 步骤 2: 安装匹配的 PyTorch 版本 (CUDA 12.1)...
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

echo.
echo 步骤 3: 安装 torch-tensorrt...
pip install torch-tensorrt

echo.
echo ========================================
echo 安装完成！
echo ========================================
echo.
echo 验证安装：
python -c "import torch; print(f'PyTorch: {torch.__version__}')"
python -c "import torchvision; print(f'TorchVision: {torchvision.__version__}')"
python -c "import torchaudio; print(f'TorchAudio: {torchaudio.__version__}')"
python -c "import torch_tensorrt; print('Torch-TensorRT: OK')" 2>nul || echo "Torch-TensorRT: 需要单独安装"

pause