VisionMamba完美复现_vision mamba复现

Vision Mamba 完美复现

本人欲魔改Mamba，在复现Vision Mamba时发现环境依赖难以搭建，经过多次尝试，现已成功。

解决关键

• conda 管理 cuda的版本conda install cudatoolkit==11.8 -c nvidia实现conda的cuda管理
• 下载对应的causal-conv1d与mamba-ssm以及对应的whl文件

解决步骤：请注意 一步都不要错

# 1. 创建自己的虚拟环境
conda create -n VisionMamba python=3.10.13
conda activate VisionMamba
# 2. cuda-11.8
conda install cudatoolkit==11.8 -c nvidia
# 3. torch torchvision torchaudio 与 官网命令一致
pip install torch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 --index-url https://download.pytorch.org/whl/cu118
# 4. nvcc
conda install -c "nvidia/label/cuda-11.8.0" cuda-nvcc
# 5. packaging
conda install packaging
# 6. causal-conv1d== & mamba-ssm==1.1.1
## 下载相应zip以及whl文件至Vim文件夹下
git clone https://github.com/hustvl/Vim.git
### mamba-ssm
https://github.com/state-spaces/mamba/releases/download/v1.1.1/mamba_ssm-1.1.1%2Bcu118torch2.1cxx11abiFALSE-cp310-cp310-linux_x86_64.whl

https://github.com/state-spaces/mamba/archive/refs/tags/v1.1.1.zip
### causal-conv1d
https://github.com/Dao-AILab/causal-conv1d/releases/download/v1.1.3/causal_conv1d-1.1.3%2Bcu118torch2.1cxx11abiFALSE-cp310-cp310-linux_x86_64.whl

https://github.com/Dao-AILab/causal-conv1d/archive/refs/tags/v1.1.3.zip

# 7. unzip & pip install
cd Vim
unzip causal-conv1d-1.1.3.zip -d ./
cd causal-conv1d-1.1.3
pip install .

cd ..

unzip mamba_ssm-1.1.1.zip -d ./
cd mamba_ssm-1.1.1
pip install .
cd ..
# 8. Vim 环境依赖
pip install -r vim/vim_requirements.txt

# 9. 更换mamaba1p1p/mamba/ 至 mamba-ssm
##  删除原环境里的mamba_ssm文件夹
rm -rf "/home/[usr_name]/miniconda/envs/VisionMamba/lib/python3.10/site-packages/mamba_ssm/"
## Vim下mamba_ssm文件夹替换进去
cp -r "/home/[usr_name]/Vim/mamba-1p1p1/mamba_ssm" "/home/[usr_name]/miniconda/envs/VisionMamba/lib/python3.10/site-packages/" 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43