训练yolov3时遇到的问题
收藏
当我想要使用GPU训练YOLOv3时遇到这个问题:
paddle框架版本:2.3
CUDA版本:10.1
cuDNN版本:7.6
一开始是发现训练的时候一直不出日志,经过调试发现用GPU训练的时候会卡在这一步:
然后如果我用CPU训练的话,这一步就不会卡住,但是别的地方会报错
想问一下怎么解决卡住的问题
0
收藏
请登录后评论