Mamba 环境安装(Ubuntu环境)

conda create -n your_env_name python=3.10.13
conda activate your_env_name
conda install cudatoolkit==11.8 -c nvidia
pip install torch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 --index-url https://download.pytorch.org/whl/cu118
conda install -c "nvidia/label/cuda-11.8.0" cuda-nvcc
conda install packaging
pip install causal-conv1d==1.1.1
pip install mamba-ssm

一般来说执行到倒数第二步就会报错了。

解决办法：

1.下载两个whl文件，注意自己的版本匹配

Release v1.0.0 · Dao-AILab/causal-conv1d · GitHub

Release v1.0.1 · state-spaces/mamba · GitHub

2.将这两个文件放在你新建的虚拟环境中的site-packages中，如果报错：is not a supported wheel on this platform，那么就改名，因为不改名的话，或者检查是不是自己的pytorch、cuda、python版本不适配，同时要注意下载的时False版本，我之前一直下载的是True版本，一直不成功。

causal_conv1d-1.0.0+cu118torch1.13cxx11abiFALSE-cp310-cp310-linux_x86_64.whl

mamba_ssm-1.0.1+cu118torch1.13cxx11abiFALSE-cp310-cp310-linux_x86_64.whl

3.最后

pip install causal_conv1d-1.0.0+cu118torch1.13cxx11abiFALSE-cp310-cp310-manylinux1_x86_64.whl
pip install mamba_ssm-1.0.1+cu118torch1.13cxx11abiFALSE-cp310-cp310-linux_x86_64.whl

4.在python交互模式下导入看看成不成功

import causal_conv1d

from mamba_ssm import Mamba