Linux下Mamba-YOLO复现

我的环境:Linux + CUDA 12.4

为了复现本项目,我前前后后花了大半周的时间,走了很多的弯路

下面这个是我自己摸索出来可以直接运行的,若有帮助,记得点个赞哦!


在 Linux 中找一个空目录,用于存放 Mamba-YOLO 项目代码

进入到该目录中,进行如下操作:

1. 克隆项目

git clone https://github.com/HZAI-ZJNU/Mamba-YOLO.git

cd Mamba-YOLO

2. 创建Conda虚拟环境

conda create -n mambayolo -y python=3.11【必须 3.8 以上】

conda activate mambayolo

3. 安装PyTorch
注意:网上教程基本都是CUDA11.8和12.1

pip3 install torch==2.3.0 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
如需检验安装,请运行下面这条稍长的命令:
python -c "import torch; print(f'PyTorch version: {torch.version}'); print(f'CUDA available: {torch.cuda.is_available()}')"

4. 安装其它必要的依赖包:

pip install seaborn thop timm einops

5. 编译安装selective_scan模块

cd selective_scan
pip install .
如遇报错:ModuleNotFoundError: No module named 'torch'
pip install .--no-build-isolation
如需检验安装,请运行下面这条稍长的命令:
python -c 'import selective_scan; print("selective_scan installed successfully!")'

cd ..

**6. 安装项目:**pip install -v -e .

**7.开始训练:**python train.py【自己写的脚本】


报错汇总:

解决:RuntimeError: GET was unable to find an engine to execute this computation

解决:ModuleNotFoundError: No module named 'torch'【文中已解决】

相关推荐
可可嘻嘻大老虎2 小时前
nginx无法访问后端服务问题
运维·nginx
JiMoKuangXiangQu2 小时前
ARM64 进程虚拟地址空间布局
linux·arm64 虚拟地址布局
阳光九叶草LXGZXJ3 小时前
达梦数据库-学习-47-DmDrs控制台命令(LSN、启停、装载)
linux·运维·数据库·sql·学习
无忧智库3 小时前
某市“十五五“地下综合管廊智能化运维管理平台建设全案解析:从数字孪生到信创适配的深度实践(WORD)
运维·智慧城市
春日见3 小时前
如何避免代码冲突,拉取分支
linux·人工智能·算法·机器学习·自动驾驶
珠海西格3 小时前
“主动预防” vs “事后补救”:分布式光伏防逆流技术的代际革命,西格电力给出标准答案
大数据·运维·服务器·分布式·云计算·能源
无垠的广袤4 小时前
【VisionFive 2 Lite 单板计算机】边缘AI视觉应用部署:缺陷检测
linux·人工智能·python·opencv·开发板
阿波罗尼亚4 小时前
Kubectl 命令记录
linux·运维·服务器
极智视界4 小时前
无人机场景 - 目标检测数据集 - 停车场停车位检测数据集下载
yolo·目标检测·数据集·无人机·voc·coco·算法训练
Fᴏʀ ʏ꯭ᴏ꯭ᴜ꯭.4 小时前
Keepalived单播模式配置与实战指南
linux·服务器·负载均衡