小白5090+cuda12.8复现vision Mamba记录

刚开始我严格按照官方readme来装依赖，甚至requirement里面的东西我都装了一遍，还是很多错，虽然说这个requirement里面的是基础依赖，但我只想先把代码跑起来，所以现在跟网上大佬们的步骤来操作，先不管requirement.txt了

一、装环境

1.创建虚拟环境, python3.13, 环境名env_vim

conda create -n env_vim python=3.13

2、pytorch，2.7

这里我选的2.7版本，因为后面还有causal_conv1d,Mamba_ssm模块要跟pytorch版本匹配，如果要按我的步骤来，建议pytorch版本也跟我一样

pip install torch==2.7.0 torchvision==0.22.0 --index-url https://download.pytorch.org/whl/cu128

3、timm， 1.0

pip install timm==1.0.11

4、mlflow==2.20.2

pip install mlflow==2.20.2

二、下载vision Mamba代码

下载vim代码压缩包后，解压缩，最好把解压缩后的项目文件夹名Vim-main改为Vim_main（不要带中间那个—符号）

解压后用PyCharm打开Vim_main, 项目结构：

不用管红框的两个模块，如果按照官方readme去安装它们的话，非常非常容易报错；det文件夹是管目标检测的，seg文件夹是管分割的，先不用看；现在只看Vim_main/vim文件夹

三、causal_conv1d 和 Mamba_ssm（我认为最关键）

下载causal_conv1d mamba_ssm

严格按照图片中红色下划线来找版本（想知道具体的，可以问AI这些红线标注的是什么意思）

2. 安装

1. 我新建了文件夹，放进去

2. 在终端进去component文件夹，执行命令：

pip install causal_conv1d-1.5.1+cu12torch2.7cxx11abiFALSE-cp313-cp313-linux_x86_64.whl

pip install mamba_ssm-2.3.0+cu12torch2.7cxx11abiFALSE-cp313-cp313-linux_x86_64.whl

3. 验证是否成功（无输出就是成功）：

python -c "from causal_conv1d import causal_conv1d_fn, causal_conv1d_update"

python -c "from mamba_ssm import Mamba, Mamba2"

四、验证模型是否能跑通

接下来就是改数据集来源，运行main.py了，参考这位大佬的博客（第三步我也看的这位大佬的，只是我用的显卡更新，cuda版本也更高）Linux上复现Vision Mamba记录_vision mamba复现-CSDN博客