windows11 wsl2 ubuntu20.04安装vision mamba并进行测试

windows11 wsl2 ubuntu20.04安装vision mamba

- - - - 安装流程
      - 使用cifar-100测试安装成功

安装流程

vision mamba安装了半天才跑通，记录一下流程
在wsl上安装cuda

wget https://developer.download.nvidia.cn/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run
sh cuda_11.8.0_520.61.05_linux.run
# 然后安装流程安装完成cuda 11.8

使用conda 创建虚拟环境，然后安装依赖库

 conda create -n mb python=3.10.13conda activate mbpip install torch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 --index-url https://download.pytorch.org/whl/cu118conda install packaging

安装最麻烦的mamba和causal_conv1d

#首先设置cudahome ,不然我的wsl会报错，找不到路径
export CUDA_HOME=/usr/local/cuda
#然后使用pip安装
pip install causal_conv1d==1.1.0
# 安装mambagit clone https://github.com/state-spaces/mamba.gitcd mamba/pip install .# 安装完成后使用vision mamba的mamba_ssm替换虚拟环境中的mamba_ssm
cd ..
git clone https://github.com/hustvl/Vim.git
cd Vim
cp -rf ./mamba-1p1p1/mamba_ssm /root  /home/zc/anaconda3/envs/mb/lib/python3.10/site-packages/

安装还没有安装的包

pip install timm mlflow -i https://pypi.tuna.tsinghua.edu.cn/simple

使用cifar-100测试安装成功

将datasets.py中的CIFAR路径和download设置好

在这里插入图片描述

运行测试程序

CUDA_VISIBLE_DEVICES=0 torchrun --master_port=6666 --nproc_per_node=1 main.py  --model vim_small_patch16_224_bimambav2_final_pool_mean_abs_pos_embed_with_midclstok_div2 --batch-size 20 --drop-path 0.05 --weight-decay 0.05 --lr 1e-3 --num_workers 1 --data-set CIFAR --data-path ./cifar-100-python --output_dir ./output/vim_small_patch16_224_bimambav2_final_pool_mean_abs_pos_embed_with_midclstok_div2 --no_amp