bigscience-workshop/petals

Allocate all GPU memory on startup

Open

#308 建立於 2023年4月24日

在 GitHub 查看
 (0 留言) (0 反應) (0 負責人)Python (424 fork)batch import
help wanted

倉庫指標

Star
 (8,248 star)
PR 合併指標
 (30 天內沒有已合併 PR)

描述

Suggested by @slush0 on Discord.

It would be great to allocate all GPU memory on startup (including max space for attention cache and max space for intermediate tensors created during backward).

Right now, the memory is allocated on-demand, and people may see OOMs only after a few days of running the server. Allocating everything right away would allows to see OOMs right away and choose a correct number of blocks before leaving a server to run in the background.

Any other way to check if we have enough GPU memory on startup would be also good.

貢獻者指南