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自带的。
5 安装mamba-ssm
可以先尝试自己编译,如果编译不成功,直接就用作者编译好的whl文件进行离线安装即可,我用的是作者编译好的whl文件