tgoop.com/awesomedeeplearning/241
Last Update:
How Much GPU Memory Needed To Server A LLM ?
This is a common question that consistnetly comes up in interview or during the disscusiion with your business stakeholders.
And it’s not just a random question — it’s a key indicator of how well you understand the deployment and scalability of these powerful models in production.
As a data scientist understanding and estimating the require GPU memory is essential.
LLM's (Large Language Models) size vary from 7 billion parameters to trillions of parameters. One size certainly doesn’t fit all.
Let’s dive into the math that will help you estimate the GPU memory needed for deploying these models effectively.
𝐓𝐡𝐞 𝐟𝐨𝐫𝐦𝐮𝐥𝐚 𝐭𝐨 𝐞𝐬𝐭𝐢𝐦𝐚𝐭𝐞 𝐆𝐏𝐔 𝐦𝐞𝐦𝐨𝐫𝐲 𝐢𝐬
General formula, 𝐦 = ((𝐏 * 𝐬𝐢𝐳𝐞 𝐩𝐞𝐫 𝐩𝐚𝐫𝐚𝐦𝐞𝐭𝐞𝐫)/𝐦𝐞𝐦𝐨𝐫𝐲 𝐝𝐞𝐧𝐬𝐢𝐭𝐲) * 𝐨𝐯𝐞𝐫𝐡𝐞𝐚𝐝 𝐟𝐚𝐜𝐭𝐨𝐫
Where:
- 𝐦 is the GPU memory in Gigabytes.
- 𝐩 is the number of parameters in the model.
- 𝐬𝐢𝐳𝐞 𝐩𝐞𝐫 𝐩𝐚𝐫𝐚𝐦𝐞𝐭𝐞𝐫 typically refers to the bytes needed for each model parameter, which is typically 4 bytes for float32 precision.
- 𝐦𝐞𝐦𝐨𝐫𝐲 𝐝𝐞𝐧𝐬𝐢𝐭𝐲 (q) refer to the number of bits typically processed in parallel, such as 32 bits for a typical GPU memory channel.
- 𝐨𝐯𝐞𝐫𝐡𝐞𝐚𝐝 𝐟𝐚𝐜𝐭𝐨𝐫 is often applied (e.g., 1.2) to account for additional memory needed beyond just storing parameters, such as activations, temporary tensors, and any memory fragmentation or padding.
𝐒𝐢𝐦𝐩𝐥𝐢𝐟𝐢𝐞𝐝 𝐅𝐨𝐫𝐦𝐮𝐥𝐚:
M = ((P * 4B)/(32/Q)) * 1.2
With this formula in hand, I hope you'll feel more confident when discussing GPU memory requirements with your business stakeholders.
#LLM
BY GenAi, Deep Learning and Computer Vision
Share with your friend now:
tgoop.com/awesomedeeplearning/241