自学内容网 自学内容网

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版本

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 


原文地址:https://blog.csdn.net/qq_34950042/article/details/142491595

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