Conda 环境配置之 -- Mamba安装（causal-conv1d、mamba_ssm 最简单配置方法）-- 不需要重新配置CDUA

以Mamba1.0为例，详细讲解最简单的Mamba配置教程！

1. 首先我们使用conda创建一个虚拟环境，命令如下：

conda create -n your_name python=3.9

2. Mamba1.0 需要 Pytorch 2.0.1 + cu117，我们在pytorch官网上使用以下命令安装：

conda install pytorch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 pytorch-cuda=11.7 -c pytorch -c nvidia

3. 这一步就比较重要了，Mamba1.0需要CUDA版本大于11.7来编译，你需要查看自己的CUDA版本是否符合，查询命令如下：

 nvcc -V

如果CUDA版本显示的是11.7或者大于11.7，如图所示：

接下来，你就可以使用以下命令安装Mamba了：

pip install causal_conv1d==1.0.0
pip install mamba_ssm==1.0.1

如果CUDA版本显示的是小于11.7，如图所示：

这种情况你首先需要检查这个虚拟环境中是否存在CUDA>=11.7，命令如下：

ls $CONDA_PREFIX/lib | grep cuda

如果是这种结果，则表明已经存在CUDA>=11.7：

如果没有，则需要重新安装 Pytorch 2.0.1 + cu117，直到出现以上结果，紧接着就只需要安装nvcc来关联虚拟环境中的CUDA，如下：

conda install -c nvidia cuda-nvcc=11.7.99

安装完毕，再运行 nvcc -V 得到如下结果：

然后就可以使用以下命令安装Mamba了，到这里你就成功了！

pip install causal_conv1d==1.0.0
pip install mamba_ssm==1.0.1

此博客纯属个人经验，欢迎评论讨论！