- 1.基础环境的安装(默认在linux+ubuntu+anaconda)具体细节可以参考我的上一篇博客https://blog.youkuaiyun.com/Sakura_zhl/article/details/144562240,其中的不同点只有causal_conv1d和mamba_ssm两个包的版本不同。
conda create -n VMamba python==3.10.13 conda activate VMamba conda install cudatoolkit==11.8 -c nvidia pip install torch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 --index-url https://download.pytorch.org/whl/cu118 conda install -c "nvidia/label/cuda-11.8.0" cuda-nvcc conda install packaging pip install mmcv==2.1.0 -f https://download.openmmlab.com/mmcv/dist/cu118/torch2.1/index.html pip install mmengine==0.10.1 pip install mmdet==3.3.0 mmsegmentation==1.2.2 mmpretrain==1.2.0 pip install causal_conv1d==1.0.0 pip install mamba_ssm==1.0.1
- 核心代码文件的创建与添加。在ultralytics/nn文件夹,建立Addmoudules文件夹,并在该文件夹下建立vmamba.py和__init__.py。打开task.py
yolov8结合vmamba,目标检测涨点起飞!
最新推荐文章于 2025-07-04 09:46:08 发布