AI算力开放-yolov8适配 mmyolo大疆无人机

先更新 cuda

1如果之前有需要更新的话需要先删除

2查看 pytorch 支持的 cuda版本

https://pytorch.org/get-started/locally/?_gl=1*1nzhx7n*_up*MQ..*_ga*NjIxOTkyMDQuMTc2Nzc2OTA4NQ..*_ga_469Y0W5V62*czE3Njc3NjkwODQkbzEkZzAkdDE3Njc3NjkwODQkajYwJGwwJGgw&__hstc=76629258.724dacd2270c1ae797f3a62ecd655d50.1746547368336.1746547368336.1746547368336.1&__hssc=76629258.9.1746547368336&__hsfp=2230748894

3查看 自己电脑 支持的cuda :nvidia-smi

4最后然后下载合适的 cuda:https://developer.nvidia.com/cuda-toolkit-archive

如有问题 参考:https://zhuanlan.zhihu.com/p/23464877518

5下载python 版本

推荐 3.10
https://www.python.org/downloads/windows/

6下载anaconda :https://www.anaconda.com/download-success

7最终环境配置:

简单 anaconda 命令大全

电脑终端中

初始化 conda init

  • 查看 Conda 版本:conda --version

  • 更新 Conda:conda update conda

  • 创建虚拟环境:conda create -n env_name python=3.8

  • 查看所有虚拟环境:conda env list

  • 激活虚拟环境:conda activate env_name

  • 退出虚拟环境:conda deactivate

  • 删除虚拟环境:conda remove --name env_name --all

    步骤 1. 创建并激活一个 conda 环境。

    复制代码
    conda create -n mmyolo python=3.10 -y
    conda activate mmyolo

    步骤 2. 安装 PyTorch。指定了torch和cuda版本

    复制代码
    pip3 install torch==2.0.0 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

    步骤 3. 验证 PyTorch 安装。正确的话,会打印版本信息和 True 字符

    复制代码
    python -c "import torch; print(torch.__version__); print(torch.cuda.is_available())"

    安装mmyolo

    步骤 0. 使用 MIM 安装 MMEngine、 MMCV 和 MMDetection 。

    复制代码
    pip install fsspec sympy==1.13.1
    pip install -U openmim
    mim install "mmengine>=0.6.0"
    mim install "mmcv>=2.0.0rc4,<2.1.0"
    mim install "mmdet>=3.0.0,<4.0.0"

    步骤 1. 安装 MMYOLO

    复制代码
    git clone https://github.com/open-mmlab/mmyolo.git
    cd mmyolo
    # Install albumentations
    mim install -r requirements/albu.txt
    # Install MMYOLO
    mim install -v -e .
    # "-v" 指详细说明,或更多的输出
    # "-e" 表示在可编辑模式下安装项目,因此对代码所做的任何本地修改都会生效,从而无需重新安装。
    复制代码
    *这里 我电脑 3060显卡  cuda 12.6 python 3.10  
    
    遇到问题1:mim告警   那就不用mim 用pip 
    问题2 找不到PyTorch  那就修改setup 文件 
    把
    注释掉 from torch.utils.cpp_extension import BuildExtension
    修改         cmdclass={'build_ext': BuildExtension}, 为 cmdclass={'build_ext': lambda: type('BuildExtension', (), {'run': lambda self: None})},
    
    先把 001-NEW-ai-inside-init.patch 放在mmyolo文件下
    git apply 0001-NEW-ai-inside-init.patch  
    
    让后修改 yolov8_s_syncbn_fast_8xb16-500e_coco.py 文件 路径放入正确的

    -----data related-----

    data_root = './data/ceshi/' # 【必须修改】数据集的根目录路径,请改为你本地的实际路径

    train_ann_file = 'annotations/instances_Train.json' # 训练集标注文件路径(相对于data_root)

    train_data_prefix = 'images/Train/' # 训练集图像文件路径前缀(相对于data_root)

    val_ann_file = 'annotations/instances_Validation.json' # 验证集标注文件路径

    val_data_prefix = 'images/Validation/' # 验证集图像文件路径前缀

    训练 :

    python tools/train.py configs/yolov8/yolov8_s_syncbn_fast_8xb16-500e_coco.py

    复制代码
相关推荐
卷无止境8 小时前
独立开发者的"富矿地带":哪些垂直领域值得你押注一辈子?
后端·python
\光辉岁月/8 小时前
5.java-数组
java·开发语言
Java后端的Ai之路8 小时前
Python进阶探索29_eval内置函数
开发语言·python·探索·eval·内置函数
计算机毕业编程指导师8 小时前
计算机毕设选题推荐:基于Hadoop与Spark的Steam游戏数据分析系统源码 毕业设计 选题推荐 毕设选题 数据分析 机器学习 深度学习
大数据·hadoop·python·计算机·spark·毕业设计·steam游戏
计算机毕业编程指导师9 小时前
【计算机毕设选题推荐】基于Hadoop+Spark的白鹿抖音评论大数据分析与可视化系统源码 毕业设计 选题推荐 毕设选题 数据分析 机器学习 深度学习
大数据·hadoop·python·计算机·spark·毕业设计·抖音评论
谢亮_vipxieliang9 小时前
Spring 事务失效的常见场景
java·开发语言·数据库·spring boot
Zootopia6269 小时前
飞行力学知识梳理1|飞行性能与稳定性
人工智能·python·算法·机器学习·无人机·学习方法·信息与通信
Y3815326629 小时前
搜索 API 延迟优化:并发数、超时与超时的真实代价
python·搜索引擎
时间的拾荒人9 小时前
Qt 多线程详解:从 QThread 到实战
开发语言·qt·面试
郑州光合科技余经理9 小时前
海外版外卖加盟:总站与分站配送规则怎么分开管
java·开发语言·前端·后端·uni-app·php·ai编程