运行Mamba项目时无法直接用pip install安装causal_conv1d和mamba_ssm
问题描述
我想运行一下Mamba但是他有依赖包causal_conv1d,直接用pip install causal_conv1d
命令安装会出错,然后我将causal_conv1d项目下载下来用python setup.py install
也不可以
解决方法
我参考了这个issue
$ git clone https://github.com/Dao-AILab/causal-conv1d.git
$ cd causal-conv1d
$ git checkout v1.1.1 # current latest version tag
$ CAUSAL_CONV1D_FORCE_BUILD=TRUE pip install .
$ cd ..
$ git clone https://github.com/state-spaces/mamba.git
$ cd mamba
$ git checkout v1.1.1 # current latest version tag
$ pip install .
但是我直接git clone会出错
(zj) biiteam@cqdx2-System-Product-Name:/media/biiteam/Storage/ZJ/mamba-main$ git clone https://github.com/Dao-AILab/causal-conv1d.git
正克隆到 'causal-conv1d'...
fatal: http://10.254.7.4/chkuser?url=github.com/Dao-AILab/causal-conv1d.git/info/refs not valid: is this a git repository?
于是我将项目下载到本地
切换到项目里面执行
CAUSAL_CONV1D_FORCE_BUILD=TRUE pip install .
成功安装
补充
安装mamba_ssm如果使用这个命令
pip install .
报错的话
Could not build wheels for mamba_ssm, which is required to install pyproject.toml-based projects
可以使用下面这个命令强制安装
MAMBA_FORCE_BUILD=TRUE pip install .
在上面我提到的issue里面的最后有人提出了这个疑问,感兴趣的可以看看。
也有可能是安装的mamba版本太高,跟自己环境的torch不匹配,可以在mamba发布的版本中查看是否有自己环境对应的torch版本
“https://github.com/state-spaces/mamba/releases”