GPU memory usage using Matlab deep learning

조회 수: 11 (최근 30일)
Amirreza Mahbod
Amirreza Mahbod 2017년 12월 12일
댓글: Amirreza Mahbod 2017년 12월 20일
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.

답변 (1개)

Joss Knight
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?
  댓글 수: 1
Amirreza Mahbod
Amirreza Mahbod 2017년 12월 20일
I am tying na d training different networks such as ResNet, VGG16 and GoogleNet

댓글을 달려면 로그인하십시오.

카테고리

Help CenterFile Exchange에서 Parallel and Cloud에 대해 자세히 알아보기

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by