项目中用到Mamba的小伙伴,causal_conv1d和 mamba-ssm两个包,但是会卡在Building wheel for mamba-ssm (setup.py) :
为了探究卡在了building的哪一步,加入–verbose进行显示:
pip install mamba-ssm --no-cache-dir --verbose
发现其会从github拉取一个wheel文件。注意:无论有没有换源都会从github拉取!
此时,我们直接去github上下载轮子whl文件,然后pip 该文件即可。
解决方案
使用魔法上网直接代理终端
先使用命令获得对应的whl文件地址,单独浏览器打开下载后放到本地再从whl进行安装。
下载causal-conv1d:
Dao-AILabcausal-conv1d
mamba-ssm:
state-spacesmamba
我的causal-conv1d 正常安装了,所以这边以mamba-ssm为例,下载后
pip install mamba_ssm-1.1.1+cu118torch2.1cxx11abiTRUE-cp310-cp310-linux_x86_64.whl
causal-conv1d文件下下载界面如下
安装成功
也可以直接拉取Docker镜像
参考:直接使用Mamba基础环境docker镜像