Mambavision installation
Ref: https://github.com/NVlabs/MambaVision/issues/86
- Docker image: pytorch/pytorch:2.5.1-cuda12.4-cudnn9-devel
- pip install causal-conv1d==1.5.0.post5
- wget https://github.com/state-spaces/mamba/releases/download/v2.2.3.post2/mamba_ssm-2.2.3.post2+cu12torch2.5cxx11abiFALSE-cp311-cp311-linux_x86_64.whl
- pip install mamba_ssm-2.2.3.post2+cu12torch2.5cxx11abiFALSE-cp311-cp311-linux_x86_64.whl
- git clone https://github.com/NVlabs/MambaVision.git
- cd MambaVision
- Replace setup.py #40 line: mamba-ssm==2.2.4" with "mamba-ssm"
- pip install -e .
댓글
댓글 쓰기