| # Core dependencies | |
| # Pin PyTorch to 2.1.0 to match CUDA 11.8 in base image (required for gsplat compatibility) | |
| torch==2.1.0 | |
| torchvision==0.16.0 | |
| einops | |
| huggingface_hub | |
| imageio | |
| numpy<2 | |
| opencv-python | |
| xformers | |
| open3d | |
| pillow | |
| omegaconf | |
| pycolmap | |
| git+https://github.com/cvg/Hierarchical-Localization.git | |
| safetensors | |
| h5py | |
| tqdm | |
| tensorboard | |
| # API server | |
| fastapi>=0.104.0 | |
| uvicorn[standard]>=0.24.0 | |
| python-multipart>=0.0.9 | |
| pydantic>=2.0.0 | |
| pydantic-settings>=2.0.0 | |
| requests>=2.31.0 | |
| boto3>=1.34.0 | |
| # Experiment tracking | |
| wandb>=0.16.0 | |
| # System monitoring | |
| psutil>=5.9.0 | |
| # Development | |
| pre-commit | |
| pytest>=7.0.0 | |