首页 Paddle框架 帖子详情
训练yolov3时遇到的问题
收藏
快速回复
Paddle框架 问答模型训练深度学习 176 0
训练yolov3时遇到的问题
收藏
快速回复
Paddle框架 问答模型训练深度学习 176 0

当我想要使用GPU训练YOLOv3时遇到这个问题:

paddle框架版本:2.3

CUDA版本:10.1

cuDNN版本:7.6

一开始是发现训练的时候一直不出日志,经过调试发现用GPU训练的时候会卡在这一步:

然后如果我用CPU训练的话,这一步就不会卡住,但是别的地方会报错

想问一下怎么解决卡住的问题

0
收藏
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户