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

相关推荐
阿拉斯攀登11 分钟前
【无人售货柜・RK+YOLO】篇 5:RK3576 部署第一步!YOLO 模型转 RKNN 全流程,新手必避的量化大坑
yolo
REDcker13 分钟前
CentOS 与主流 Linux 发行版历史与版本综述
linux·centos·numpy
劳埃德福杰13 分钟前
【Kylin银河麒麟】文件系统磁盘空间满导致无法进入操作系统
运维·服务器·电脑·笔记本电脑·kylin
逻辑峰22 分钟前
ReadStat在Linux的安装和使用
linux·运维·服务器
Lsir10110_33 分钟前
【Linux】序列化与反序列化——网络计算器的实现
linux·运维·网络
脆皮的饭桶1 小时前
给负载均衡做高可用的工具Keepalived
运维·服务器·负载均衡
袁小皮皮不皮1 小时前
【HCIA】第三章TCP/IP协议栈中其他主要协议
运维·服务器·网络·网络协议·tcp/ip
阿梦Anmory1 小时前
保姆级教程:Flask应用实现后台常驻运行(Linux服务器)
linux·服务器·flask
夏日听雨眠1 小时前
Linux学习1
linux·服务器·学习
小生不才yz1 小时前
【Makefile 专家之路 | 函数篇】10. 逻辑增强:逻辑函数(if/and/or)与环境断言(info/error)
linux