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

相关推荐
Coding茶水间3 小时前
基于深度学习的肾结石检测系统演示与介绍(YOLOv12/v11/v8/v5模型+Pyqt5界面+训练代码+数据集)
图像处理·人工智能·深度学习·yolo·目标检测·机器学习·计算机视觉
C语言不精4 小时前
Tina Linux SDK编译SDK-linux环境下实现
linux·运维·服务器
WG_174 小时前
Linux:进程地址空间
linux
番茄迷人蛋4 小时前
后端项目服务器部署
java·运维·服务器·spring
LILR_4 小时前
简单学docker
运维·docker·容器
Blossom.1184 小时前
基于多模态大模型的工业质检系统:从AOI到“零样本“缺陷识别的产线实践
运维·人工智能·python·机器学习·自动化·测试用例·知识图谱
Xの哲學4 小时前
Linux MAC层实现机制深度剖析
linux·服务器·算法·架构·边缘计算
奇点爆破XC4 小时前
centos进阶命令.Linux系统介绍(运维版)
linux·运维·centos
南棱笑笑生4 小时前
20251215给飞凌OK3588-C开发板适配Rockchip原厂的Buildroot【linux-5.10】后调通typeC1接口
linux·c语言·开发语言·rockchip