【问题解决】2024-1-9 用gpu跑yolov8模型遇到的问题 : An attempt has been made to start a new process before the curre

本文围绕使用GPU训练YOLO模型时运行mytrain.py文件报错展开。报错信息显示试图在当前进程完成引导阶段前启动新进程,分析指出是Windows系统使用多进程模块未正确设置条件所致。给出解决方案,需在脚本中添加if __name__ == '__main__'条件判断。

问题描述:An attempt has been made to start a new process before the current process has finished its bootstrapping phase.

想用gpu训练模型,运行mytrain.py

mytrain.py文件:
from ultralytics import YOLO
import multiprocessing
# 加载一个模型
    model = YOLO('yolov8n.yaml')  # 从yaml构建一个新模型
    model = YOLO('weights/yolov8n.pt')  #  加载预先训练的模型(建议进行训练)
    model = YOLO('yolov8n.yaml').load('weights/yolov8n.pt')  # 根据YAML构建并传递权重
    results = model.train(data='datas/my_data.yaml', epochs=100, imgsz=640)

报错:

canning D:\myDesktop\ultralytics-main2\datas\labels\train.cache... 1720 images, 0 backgrounds, 0 corrupt: 100%|██████████| 1720/1720 [00:00<?, ?it/s] Traceback (most recent call last): File "<string>", line 1, in <module> File "D:\MyTools\mylab\lib\multiprocessing\spawn.py", line 116, in spawn_main exitcode = _main(fd, parent_sentinel) File "D:\MyTools\mylab\lib\multiprocessing\spawn.py", line 125, in _main prepare

评论 4
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值