笔者近期需要使用mamba-ssm,但是由于第一次安装,加上网上关于mamba安装比较少,所以走了很多弯路,浪费了许多时间。所以我想写出我的经验,供大家参考。
我的建议就是静态下载,直接导入文件,理由就是最低限度减少torch和cuda和py的版本错误,因为由于文件名字上存在这些版本,所以你在下载的时候就会去确认安装的这些版本是否匹配而不是完全的在试错。第二就是可以看到他下载的进度,如果在终端上使用conda pip mamba-ssm的话,在他下载轮子的时候就纯黑盒,也很难知道是否会存在问题。
安装过程:
1、确认自己torch+cuda+py的版本:torch2.3.1+cu118+py3.12
2、进入https://github.com/Dao-AILab/causal-conv1d/releases?page=3和https://github.com/state-spaces/mamba/releases?page=1两个链接下载对应的文件(cu11和cu118不一样,一定要注意,但是torch可以下载2.3的)
3、放入文件夹下,直接pip install 完整文件名(先causal-conv1d后mamba-ssm)
4、验证一下,让mamba跑一个最简单的前向,看看有没有什么输出问题。