2025年9月19日 vision mamba linux系统成功安装(全程记录)
参考:Mamba环境配置教程(无脑式离线安装&包含Mamba2)_mamba2安装-CSDN博客
感谢博主分享,我整体流程就是按照这个博主来的,所有的版本和该博主完全对应
1、首先创建环境:
conda create -n vm python=3.10
2、激活环境
conda activate vm
3、安装torch
(和该博主不同,我直接复制的pytorch官网的pip,还是很快的,2-3分钟就能装好,不过有的人很慢的话也可以去从博主中的百度网盘下载链接进行本地安装)
pip install torch==2.3.1 torchvision==0.18.1 torchaudio==2.3.1 --index-url https://download.pytorch.org/whl/cu118
4、装triton和transformers库(这里很奇怪,我不加源就特别慢!!!,加了源之后就特别快)
pip install triton==2.3.1 -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install transformers==4.43.3 -i https://pypi.tuna.tsinghua.edu.cn/simple
5、原博主说装一下cuda-nvcc这个库,我死活装不上。。。干脆没装,想继续往下看看试试运气,看看能不能跑通,结果后面还真给我跑通了(这里小小的吐槽一下,不知道是我网的问题还是服务器问题,conda怎么都不行,挂vpn也不行)
6、最后安装 causal-conv1d 和mamba-ssm库,这里我也是按照博主网盘链接里面的两个whl文件进行pip install 安装的
pip install causal_conv1d-1.4.0+cu118torch2.3cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
pip install mamba_ssm-2.2.2+cu118torch2.3cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
7、安装完后在输入框内输入进行运行测试:
python /home/cr7733/Vim-main/vim/main.py --model vim_tiny_patch16_224_bimambav2_final_pool_mean_abs_pos_embed_with_midclstok_div2 --batch-size 128 --drop-path 0.0 --weight-decay 0.1 --num_workers 1 --data-path /home/cr7733/mmpretrain-main/data/imagenet --output_dir ./output/vim_tiny_patch16_224_bimambav2_final_pool_mean_abs_pos_embed_with_midclstok_div2 --no_amp
8、报错:
ModuleNotFoundError: No module named 'timm'
然后同样pip install一下,也是装不上,加清华源
9、继续运行6的指令:
还是报错:
ModuleNotFoundError: No module named 'mlflow'
缺少库的报错是最简单的报错,继续pip就行了
10、继续运行指令,报错:
TypeError: Mamba.__init__() got an unexpected keyword argument 'bimamba_type'
这个报错,又去别的博主那边搜了一下,
vision mamba 运行训练记录,解决bimamba_type错误-CSDN博客
按照这个博主说的“官方项目里的mamba_ssm替换安装在conda环境里的mamba_ssm”
尝试一下,
cp -rf mamba-1p1p1/mamba_ssm /home/liyhc/anaconda3/envs/mamba/lib/python3.10/site-packages
11、继续运行,成功!!!
写一下感想:折腾了一下午,本来想找点论文baseline的,一看这个vision mamba挺火的,上谷歌学术一搜25年的文章,很多顶刊都在发,我寻思整一个跑下来看看。一开始是跟着github的官方流程来,结果根本行不通,pip根本跑不起来,然后按照我自己的经验,我想肯定需要本地安装了,就自己尝试了一下本地安装,依旧报错
这个时候我就开始在csdn上找各种教程,我感觉很少,并且说的都不是很全,除了我引用的第一个博主,我跟着他成功了,其他的都是五花八门的。寻思咸鱼花点钱找人给装一下,应该也要不了多少钱,结果说30块钱只装一个mamba_ssm。。。想钱想疯了吧,我自己还被一顿嘲讽。。。
晚上想着再装一下试试,结果还真被我跑通了,在这里写一下我成功的案例供大家参考,祝科研顺利
更多推荐
所有评论(0)