padelGan视频超分模型GPU内存不足,如何解决?
收藏
按照官方的debug文档,还是没有解决 config配置文档在详细一点就好了
---------------------------------------
Out of memory error on GPU 0. Cannot allocate 1.439175GB memory on GPU 0, 22.777100GB memory has been allocated and available memory is only 932.562500MB.
Please check whether there is any other process using GPU 0.
1. If yes, please stop them, or start PaddlePaddle on another GPU.
2. If no, please decrease the batch size of your model.
If the above ways do not solve the out of memory problem, you can try to use CUDA managed memory. The command is `export FLAGS_use_cuda_managed_memory=false`.
(at /paddle/paddle/fluid/memory/allocation/cuda_allocator.cc:87)
. (at /paddle/paddle/fluid/imperative/tracer.cc:307)
1
收藏
请登录后评论
减小 batch size 试一试吧
如前所述,尝试调整批量大小的内存消耗
运行环境是什么配置,显存多大?现在AIStudio支持128G显存了,只不过算力卡有亿点点贵
视频模型size已经最小了 是不是我视频太长的原因?
你这3090吧=。=
batchsize设置成1
分辨率缩小点
正常,减batchsize或者上大服务器
可以考虑切分为单图片超分后拼成视频