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'【文中已解决】

相关推荐
꧁执笔小白꧂4 小时前
详细pyenv+venv+pip安装ultralytics
yolo·venv·pyenv·ultralytics
大耳朵-小飞象7 小时前
电力安全运维的智能密码:BACS如何破解设备全生命周期管理难题,让电网安全“看得见、管得住”?
运维·安全·智慧城市·能耗系统·楼宇智控·未来生活
极客侃科技7 小时前
制造企业 MES/APS 选型:SAP PP/DS 集成、ERP-MES 边界划分与一体化架构要点
运维·架构·制造
HLC++8 小时前
Linux的进程间通信
android·linux·服务器
华清远见IT开放实验室9 小时前
实验室建设案例 | 石家庄科技信息职业学院嵌入式实验室——从底层硬件到系统应用,一所应用型高校的嵌入式人才培养这样落地
linux·arm开发·stm32·嵌入式硬件·高校·实验室建设
生活爱好者!10 小时前
我把NAS当作下载机,docker一键部署qb
运维·docker·容器
groundhappy12 小时前
idalib安装和codex ida-mcp配置
linux·开发语言·python
尽兴-12 小时前
企业业务系统架构选型与渐进式演进
运维·系统架构·devops
通信小小昕12 小时前
Ubuntu 26.04 中文输入法安装
linux·运维·ubuntu
壮哥_icon13 小时前
【Android 系统开发】使用 BAT 脚本高效自动化管理 /system/priv-app/ 系统应用(安装与卸载)
android·运维·自动化