GPU memory usage using Matlab deep learning
11 vues (au cours des 30 derniers jours)
Afficher commentaires plus anciens
Hi, I am using Matlab for fine tuning pre-trained CNN. The code works without any problems. However, when I monitor my GPU memory usage lively (using both GPU-Z and nvidia-smi in command line), I notice it varies between 4-7 GB. My GPU memory is 8 GB (GTX1070). Have any idea what is the reason behind that? The total running time is around 30 minutes for my code. I also observe much less running time on Ubuntu operation system when I tried to run the similar code on Keras.
0 commentaires
Réponses (1)
Joss Knight
le 19 Déc 2017
While you are training a network it needs a lot of working memory. The deeper the network, the more memory it needs. Which network are you training?
Voir également
Catégories
En savoir plus sur Parallel and Cloud dans Help Center et File Exchange
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!