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

张开发
• 2026/4/15 4:52:10 • 15 分钟阅读

最新文章

推荐文章

相关文章

分享文章

小白5090+cuda12.8复现vision Mamba记录
刚开始我严格按照官方readme来装依赖甚至requirement里面的东西我都装了一遍还是很多错虽然说这个requirement里面的是基础依赖但我只想先把代码跑起来所以现在跟网上大佬们的步骤来操作先不管requirement.txt了一、装环境1.创建虚拟环境, python3.13, 环境名env_vimconda create -n env_vim python3.132、pytorch2.7这里我选的2.7版本因为后面还有causal_conv1d,Mamba_ssm模块要跟pytorch版本匹配如果要按我的步骤来建议pytorch版本也跟我一样pip install torch2.7.0 torchvision0.22.0 --index-url https://download.pytorch.org/whl/cu1283、timm 1.0pip install timm1.0.114、mlflow2.20.2pip install mlflow2.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.1cu12torch2.7cxx11abiFALSE-cp313-cp313-linux_x86_64.whlpip install mamba_ssm-2.3.0cu12torch2.7cxx11abiFALSE-cp313-cp313-linux_x86_64.whl3. 验证是否成功无输出就是成功python -c from causal_conv1d import causal_conv1d_fn, causal_conv1d_updatepython -c from mamba_ssm import Mamba, Mamba2四、验证模型是否能跑通接下来就是改数据集来源运行main.py了参考这位大佬的博客第三步我也看的这位大佬的只是我用的显卡更新cuda版本也更高Linux上复现Vision Mamba记录_vision mamba复现-CSDN博客

更多文章