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 小时前
Nginx 缓存命中率监控:从 X-Cache-Status 到实时统计
运维·nginx·缓存
Apipi*4 小时前
30天速通Linux 第六章信号及信号处理
linux·运维·信号处理
天启HTTP5 小时前
爬虫防封核心原理:IP轮换机制与风控规避逻辑
linux·服务器·网络·爬虫·tcp/ip
运维行者_6 小时前
网络性能监控怎么做?从自动发现到根因分析的4个环节
运维·服务器·网络·人工智能·支持向量机
此时不提桶,更待何时6 小时前
06-14-A-Kafka集群运维与迁移实战详解
运维·kafka
广州浮点FLOATLIC8 小时前
许可证服务器迁移后软件打不开:研发 IT 怎样定位连接问题
linux·服务器·数据库
IT研究所8 小时前
AI-ITR平台如何减少客户问题反复升级?
大数据·运维·人工智能·低代码·自然语言处理·安全架构·企微
BLUcoding8 小时前
接口服务公网超时排查记录:SecureRandom 阻塞问题分析与解决
java·linux·springboot·aes·securerandom
AIgorithmGEEK8 小时前
[Linux]从手写报头到内核套路:序列化、反序列化与自定义协议全链路
linux·运维·服务器·网络·序列化·反序列化
Nil2088 小时前
leetcode 139单词拆分
linux·运维·服务器