tensorflow-YOLO1 icon indicating copy to clipboard operation
tensorflow-YOLO1 copied to clipboard

机器配置问题

Open TianshengSun opened this issue 5 years ago • 7 comments

请问我可以用GTX1650 4G 8G运行内存的笔记本运行该程序吗,为什么我设置成GPU运行在训练的刚开始的时候就会导致运行内存很快达到8G然后进程被自动结束?

TianshengSun avatar Dec 16 '19 13:12 TianshengSun

Hi @Huntersts 也许你可以尝试减小batchsize 我当时的环境是1080TI,显存是11G 如果做深度学习,显存最好8G(个人推荐,可不采纳)。尽量薅羊毛

LeslieZhoa avatar Dec 16 '19 15:12 LeslieZhoa

那请问一下为什么我这样做会导致机器的运行内存暴涨啊 我设置batchsize为1然后限制了GPU显存使用的大小(萌新)

TianshengSun avatar Dec 16 '19 15:12 TianshengSun

@Huntersts 请问你限制了百分比吗?是多少。你是运行内存超了还是显存超了呢?

LeslieZhoa avatar Dec 16 '19 15:12 LeslieZhoa

限制在70% 实际运行时大约在2G左右 然后运行内存很快就从30%涨到100% 然后进程就结束了

TianshengSun avatar Dec 16 '19 15:12 TianshengSun

运行内存,很可能是数据读取存储的时候爆掉了,还是建议用至少1080TI来进行深度学习

LeslieZhoa avatar Dec 16 '19 15:12 LeslieZhoa

@Huntersts 萌新的话不建议进入图像的领域,除非你特别喜欢,工作不是很好找,除非你技能过硬。 nlp,推荐算法,还是很火爆的

LeslieZhoa avatar Dec 16 '19 15:12 LeslieZhoa

多谢大佬

TianshengSun avatar Dec 16 '19 15:12 TianshengSun