- torch>=2.1.0
- torchvision>=0.3
- numpy
- opencv-python
- matplotlib
- kornia>=0.6.11
- # mambaglue/mambaglue.py 直接依赖(原仓库未写全)
- omegaconf
- einops
- requests
- # mamba-ssm:PyPI 无 Windows 官方轮子。用 curl 下载见项目 vendor/mamba-ssm-wheels/*.cmd
- # - FuouM(HF,cp311):https://huggingface.co/FuouM/mamba-ssm-windows-builds/tree/main
- # - loscrossos(cp312,需匹配 torch/cuda 标签):https://github.com/loscrossos/lib_mamba/releases
- # - Linux:pip install mamba-ssm 或 GitHub Releases 的 manylinux 轮子
|