Join the conversation

Join the community of Machine Learners and AI enthusiasts.

Sign Up
AlexBodner 
posted an update Aug 30
Post
1586
💾🧠Want to know how much VRAM you will need for training your model? 💾🧠
Now you can use this app in which you can input a torch/tensorflow summary or the parameters count and get an estimate of the required memory!
Use it in: howmuchvram.com

Also, everything is Open Source so you can contribute in repo: https://github.com/AlexBodner/How_Much_VRAM
Leave it a star⭐