如何在ubuntu配置Vision-Mamba环境

本人服务器端环境

ubuntu22.04、Anaconda(cuda 11.8,python 3.11)（cuda版本必须为11.8以上，低于11.8的mamba不支持，同时python环境必须为3.10以上的版本）

一定要特别注意，安装的环境一定要为base环境，如果不是base环境，按照下面的步骤操作后只能通过命令行并且只能以root用户下运行代码，不能在其他非root用户下运行代码，同时也没法借助vscode、pycharm等编辑器进行代码调试，会出现以下报错：
TypeError: Mamba.init() got an unexpected keyword argument ‘bimamba_type’

安装步骤

步骤一

从官网下载Vision-Mamba代码（git方式或者直接下载解压均可）

步骤二

将conda环境切换到base环境下，然后运行以下代码安装torch（由于我的版本为cuda11.8的，所以torch安装为以下命令，各位如果版本更高的话可以直接去Pytorch官网选择合适版本安装）

pip install torch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 --index-url https://download.pytorch.org/whl/cu118

步骤三

切换到Vision-Mamba目录下，安装Vision-Mamba依赖包，运行以下安装命令即可

pip install -r vim/vim_requirements.txt

步骤四（关键步骤）

安装causal_conv1d ，安装causal_conv1d时运行以下命令即可，注意安装版本一定要为1.1.1的，版本太高的话会不适配，会出现以下错误：
causal_conv1d_fwd(): incompatible function arguments. The following argument types are supported:
1. (arg0: torch.Tensor, arg1: torch.Tensor, arg2: Optional[torch.Tensor], arg3: Optional[torch.Tensor], arg4: Optional[torch.Tensor], arg5: Optional[torch.Tensor], arg6: bool) -> torch.Tensor

安装代码如下：

pip install causal-conv1d==1.1.1

如果直接用pip install causal_conv1d==1.1.1命令安装出错的话，可以按照如下操作：首先切换到Vision-Mamba目录下，然后运行以下命令

cd causal_conv1d
CAUSAL_CONV1D_FORCE_BUILD=TRUE pip install .

步骤五（关键步骤）

安装mamba-ssm，按照Vision-mamba安装会出现以下错误：

TypeError: Mamba.init() got an unexpected keyword argument ‘bimamba_type’

安装时首先从mamba官网将依赖包下载下来，通过使用以下命令进行安装：

git clone https://github.com/state-spaces/mamba.git
cd mamba
pip install .

或者直接下载解压后运行以下命令：

cd mamba
pip install .

接下来切换到Vision-Mamba/mamba-1p1p1目录下，将该目录下的mamba-ssm跟base环境中的mamba-ssm进行替换，命令如下：

cd Vision-Mamba/mamba-1p1p1
cp -rf mamba_ssm /xxx/anaconda3/lib/python3.xx/site-packages

注意第二行代码要改为自己环境的路径
执行完上面的步骤mamba就可以正常使用了，以下是在pycharm中正常正常运行截图：

如果各位小伙伴在安装过程中有任何问题，欢迎与我交流