# windows目前装不了
目录
安装步骤:
可能出现的问题:
安装步骤:
需要的配置:
CUDA(11.6+)
pytorch (1.12+) 先装好和CUDA版本对应的torch!!不然后面问题一堆
packaging
triton
官方readme.md文件要求安装的:
causal-conv1d(>=1.2.0)
whl文件下载地址:https://github.com/Dao-AILab/causal-conv1d/releases)
mamba-ssm (whl下载地址:https://github.com/state-spaces/mamba/releases)
这两个包在安装过程中非常容易出现 building wheel fail(装好conv1d之后偶尔可以直接 pip install mamba-ssm)为了省事建议下载whl文件(注意选择和你的cuda, torch以及python版本对应的。abiTRUE/FALSE 不知道啥意思,目前按abiFALSE装的)并本地pip安装。
cd path_of_the_whl_file #进入这两个whl文件所在目录
pip install packagename.whl #packagename 替换成你下载的文件名
安装完成后 运行官方代码中给出的 setup.py 来搭建必要的环境
cd path_of_mamba-main #进入mamba-main所在目录
python setup.py build
python setup.py install
可能出现的问题:
1. pip install causal-conv1d 和 mamba-ssm 出现 building wheels fail:
不知道为什么。解决方法就是下载whl,然后pip安装
2.装好包,运行代码时出现
libcudart.so.10.0: cannot open shared object file: No such file or directory
CUDA版本不匹配。检查一下causal-conv1d 和 mamba-ssm 这两个包和pytorch、CUDA版本是否匹配。
解决方法:重装causal-conv1d,mamba-ssm,然后build, install