Windows 下安装mamba_ssm 记录，包括causal-conv1d和mamba-ssm

news2024/10/8 14:20:14

Windows 下安装mamba_ssm 记录

- 1 重要参考文献
- 2 具体安装步骤
- 3 一些提醒事项
- 4 安装causal-conv1d
- 5 安装mamba-ssm
- 6 结果展示

1 重要参考文献

Window 下Mamba 环境安装踩坑问题汇总及解决方法
在这里插入图片描述

2 具体安装步骤

重点看的是这篇 Window 下Mamba 环境安装踩坑问题汇总及解决方法

我也是装了两天左右才把windows的mamba-ssm装好，摸索了很多，下面是一些在安装过程中的问题和提示，仅供参考，建议先从头到尾过一遍，然后再跟着原始博客步骤安装。

我最后采取的所有办法都是离线安装，下载了作者提供的causal-conv1d和mamba-ssm。

现在回过头看，只需要将以下whl文件准备好，新建一个python10的环境，然后pip install 离线安装即可：
在这里插入图片描述

torch环境准备：
numpy-1.24.1-cp310-cp310-win_amd64.whl
torch-2.1.1+cu118-cp310-cp310-win_amd64.whl
torchaudio-2.1.1+cu118-cp310-cp310-win_amd64.whl
torchvision-0.16.1+cu118-cp310-cp310-win_amd64.whl
causal_conv1d准备
causal_conv1d-1.1.1-cp310-cp310-win_amd64.whl
mamba_ssm 准备
mamba_ssm-1.1.3-cp310-cp310-win_amd64.whl

以下的安装过程都是我离线安装出现的问题，仅供参考
以下的安装过程都是我离线安装出现的问题，仅供参考
以下的安装过程都是我离线安装出现的问题，仅供参考

3 一些提醒事项

在这里插入图片描述

1. 必须确保python3.10, 然后安装的torch是cuda版本
2. 如果安装torch，torchvision torchaudio太慢了，建议直接去网址上找对应版本下载然后离线下载即可（有时候用镜像源下载可能会下载成cpu版本）
在这里插入图片描述
下载链接：https://download.pytorch.org/whl/cu118
进去直接ctrl+F进行搜素即可：
numpy-1.24.1-cp310-cp310-win_amd64.whl
torch-2.1.1+cu118-cp310-cp310-win_amd64.whl
torchaudio-2.1.1+cu118-cp310-cp310-win_amd64.whl
torchvision-0.16.1+cu118-cp310-cp310-win_amd64.whl
3. 安装结束使用指令看一下版本号是否对应
在这里插入图片描述

4 安装causal-conv1d

在这里插入图片描述
可以先尝试自己编译，如果编译不成功，直接就用作者编译好的whl文件进行离线安装即可
报错：
我这边报错的都是和“setuptools”有关后面，参考博客：歪打正着解决 ModuleNotFoundError: No module named ‘setuptools.command.build‘ 问题
这个报错总结起来就是pip uninstall setuptools，把setuptools卸载干净就行，包括python自带的。