我的环境: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'【文中已解决】