CUDA is being reinitialized for each sub processes which is training a keras model
I was pursuing a machine learning certification course and entered a problem and could not resolve it. I am using CUDA to train one after one on many tensorflow keras models on my Graphic processing unit. At first the program started to crash stating that the system is out of memory. I asked my course instructor and he informed that using a new sub-process for every model would clear the GPU memory automatically and it did work. But the thing is every each process reinitialize CUDA making it very time consuming. Although I asked my course instructor about the same, but thought of asking here, anybody know about it?

댓글 수: 1

Joss Knight
Joss Knight 2020년 12월 11일
Are you asking a question about python and TensorFlow in a MATLAB forum?

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

답변 (0개)

카테고리

도움말 센터File Exchange에서 Deep Learning Toolbox에 대해 자세히 알아보기

질문:

2020년 12월 9일

댓글:

2020년 12월 11일

Community Treasure Hunt

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

Start Hunting!

Translated by