自学内容网 自学内容网

mamba-ssm安装卡着不动

项目中用到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镜像


原文地址:https://blog.csdn.net/qq_43767886/article/details/138681486

免责声明:本站文章内容转载自网络资源,如本站内容侵犯了原著者的合法权益,可联系本站删除。更多内容请关注自学内容网(zxcms.com)!