deeplabv3-plus-pytorch icon indicating copy to clipboard operation
deeplabv3-plus-pytorch copied to clipboard

Published 20 hours ago • verified publisher icon bubbliiiing

运行到51步，跳GPU运行内存不足。batch已经降到2了。

Open rain0322 opened this issue 1 year ago • 6 comments

该如何调整，调整图片大小嘛？

rain0322 avatar Mar 19 '23 15:03 rain0322

解决了吗 @rain0322 我也是同样的问题

ngz-sun avatar Mar 27 '23 07:03 ngz-sun

batch降低

bubbliiiing avatar Mar 27 '23 16:03 bubbliiiing

RuntimeError: CUDA out of memory. Tried to allocate 1.25 GiB (GPU 0; 6.00 GiB total capacity; 1.09 GiB already allocated; 2.85 GiB free; 1.34 GiB reserved in total by PyTorch) 我的直接就运行不起来， image

t973288913 avatar Mar 28 '23 04:03 t973288913

num_workers 用于设置是否使用多线程读取数据，1代表关闭多线程

#                   开启后会加快数据读取速度，但是会占用更多内存
#                   keras里开启多线程有些时候速度反而慢了许多
#                   在IO为瓶颈的时候再开启多线程，即GPU运算速度远大于读取图片的速度。
#------------------------------------------------------------------#
num_workers         = 1

关闭多线程成功跑起来了

t973288913 avatar Mar 28 '23 04:03 t973288913

解决了吗 @rain0322 我也是同样的问题

解决了，调小图片大小。例如我之前是932，1163左右的。统一取短边。932，932

rain0322 avatar Mar 30 '23 03:03 rain0322

num_workers 用于设置是否使用多线程读取数据，1代表关闭多线程

#                   开启后会加快数据读取速度，但是会占用更多内存
#                   keras里开启多线程有些时候速度反而慢了许多
#                   在IO为瓶颈的时候再开启多线程，即GPU运算速度远大于读取图片的速度。
#------------------------------------------------------------------#
num_workers         = 1

关闭多线程成功跑起来了

多谢

rain0322 avatar Mar 30 '23 03:03 rain0322