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 .

댓글

이 블로그의 인기 게시물

sklearn tsne + matplotlib scatter

cp with progress / verify