不要按照vim本身的安装方式,python3.10用下面的方式下载whl安装,然后将官方的mamba_ssm替换掉就行了
wget https://github.com/Dao-AILab/causal-conv1d/releases/download/v1.1.3.post1/causal_conv1d-1.1.3.post1+cu118torch2.1cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
wget https://github.com/state-spaces/mamba/releases/download/v1.1.1/mamba_ssm-1.1.1+cu118torch2.1cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
pip install causal_conv1d-1.1.3.post1+cu118torch2.1cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
pip install mamba_ssm-1.1.1+cu118torch2.1cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
具体细节
#pip install torch==2.1.2 torchvision==0.16.2 torchaudio==2.1.2 --index-url https://download.pytorch.org/whl/cu118
timm==0.4.12
## 用conda env list 查看刚才安装的mamba环境的路径,我的mamba环境在/opt/miniconda3/envs/mamba_ocr_310
#conda env list
#用官方项目里的mamba_ssm替换安装在conda环境里的mamba_ssm
#cp -rf mamba-1p1p1/mamba_ssm /opt/miniconda3/envs/mamba_ocr_310/lib/python3.10/site-packages