mamba-ssm安装说明(Ubuntu)

1、直接使用pip install mamba-ssm,存在的问题(undefined symbol: _ZN3c104cuda9SetDeviceEi in import selective_scan_cuda)

猜测原因:没有安装causal_conv1d

2、解决办法——首先下载whl文件,然后pip install **whl文件

(1)从Releases · Dao-AILab/causal-conv1d · GitHub中选择对应的causal_conv1d

其中cu118表示cuda版本至少大于等于11.8;cxx11abiTrue和cxx11abiFALSE表示关于C++ ABI的标志,cxx11abiFALSE兼容性更好,一般选择它;cp310表示Python 3.10

(2)从Releases · state-spaces/mamba · GitHub中选择对应的mamba-ssm版本

(3)安装指令

pip install causal_conv1d-1.4.0+cu118torch2.3cxx11abiFALSE-cp38-cp38-linux_x86_64.whl  # 优先安装它;然后安装mamba-ssm

pip install mamba_ssm-2.2.2+cu118torch2.3cxx11abiFALSE-cp38-cp38-linux_x86_64.whl

备注:我的设备是Ubuntu 20.04,Python 3.8,cuda 11.8 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值