Linux中Mamba的有效安装

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