GPU memory usage using Matlab deep learning
조회 수: 11 (최근 30일)
이전 댓글 표시
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
답변 (1개)
Joss Knight
2017년 12월 19일
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?
참고 항목
카테고리
Help Center 및 File Exchange에서 Parallel and Cloud에 대해 자세히 알아보기
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!