当前位置:   article > 正文

yolov8 ultralytics库实现多机多卡DDP训练_yolo8多机多卡

yolo8多机多卡

参考:
https://github.com/ultralytics/ultralytics/issues/6286

ddp训练报错,问题修改:
https://blog.csdn.net/weixin_41012399/article/details/134379417

RuntimeError: CUDA error: invalid device ordinal CUDA kernel errors
might be asynchronously reported at some other API call, so the
stacktrace below might be incorrect. For debugging consider passing
CUDA_LAUNCH_BLOCKING=1. Compile with TORCH_USE_CUDA_DSA to enable
device-side assertions.

主要修改机器二上代码即可(主节点不用修改):
1)/root/miniconda3/envs/up/lib/python3.9/site-packages/ultralytics/engine/trainer.py

from ultralytics.utils import (DEFAULT_CFG, LOGGER, RANK
    声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/笔触狂放9/article/detail/658892
    推荐阅读
    相关标签
      

    闽ICP备14008679号