項目中用到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鏡像