| # Base Python packages | |
| numpy<=1.26.4 | |
| scipy | |
| addict | |
| timm | |
| psutil | |
| huggingface_hub | |
| opencv-python-headless | |
| einops | |
| # PyTorch and CPU version for torch 2.5.0 | |
| --find-links https://data.pyg.org/whl/torch-2.5.0+cpu.html | |
| torch==2.5.0 | |
| torchvision==0.20.0 | |
| torchaudio==2.5.0 | |
| # Extra pip dependencies that depend on PyTorch | |
| torch-scatter | |
| spconv | |
| # flash-attn | |
| # visualization | |
| trimesh | |
| camtools | |
| open3d | |
| matplotlib | |
| # Optional: If you need `ninja` as a build tool | |
| ninja | |
| # # Concerto (local package) | |
| # -e . |