Mamba 环境安装：causal-conv1d和mamba-ssm报错解决办法

JOYCE_Leo16

发布于 2024-03-19 17:07:55

2.2K0

发布于 2024-03-19 17:07:55

文章被收录于专栏：计算机视觉计算机视觉

问题描述：

在执行命令 pip install causal_conv1d 和 mamba_ssm 出错：

解决方案：

1、使用网友配置好的Docker环境，参考：解决causal_conv1d和mamba_ssm无法安装 -＞直接使用Mamba基础环境docker镜像 DockHub仓库地址：https://hub.docker.com/repository/docker/kom4cr0/cuda11.7-pytorch1.13-mamba1.1.1/general 代码：docker pull kom4cr0/cuda11.7-pytorch1.13-mamba1.1.1:1.1.1

2、直接下载工程文件，再setup。具体可参考：运行Mamba项目时无法直接用pip install安装causal_conv1d和mamba_ssm_pip install causal-conv1d编译文件-CSDN博客

（笔者依然未安装成功，但是原作者以及GitHub issue 里有部分人可以安装成功）

参考步骤为：

git clone https://github.com/Dao-AILab/causal-conv1d.git
cd causal-conv1d
git checkout v1.1.1 # current latest version tag
CAUSAL_CONV1D_FORCE_BUILD=TRUE pip install .
cd ..
git clone https://github.com/state-spaces/mamba.git
cd mamba
git checkout v1.1.1 # current latest version tag
pip install .

3、受博文 “flash-attention踩坑：使用conda管理CUDA”启发，合理调整按照顺利，先安装CUDA，并且安装cuda-nvcc，正确的安装步骤如下：

conda create -n your_env_name python=3.10.13
conda activate your_env_name
conda install cudatoolkit==11.8 -c nvidia
pip install torch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 --index-url https://download.pytorch.org/whl/cu118
conda install -c "nvidia/label/cuda-11.8.0" cuda-nvcc
conda install packaging
pip install causal-conv1d==1.1.1
pip install mamba-ssm

参考：Mamba 环境安装踩坑问题汇总及解决方法_安装 causal-conv1d-CSDN博客

本文参与?腾讯云自媒体分享计划，分享自作者个人站点/博客。

原始发表：2024-03-19，如有侵权请联系 cloudcommunity@tencent.com 删除

解决方案