Error on using Tenser flow.

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 Kommentar

Joss Knight
Joss Knight am 11 Dez. 2020
Are you asking a question about python and TensorFlow in a MATLAB forum?

Melden Sie sich an, um zu kommentieren.

Antworten (0)

Kategorien

Mehr zu Deep Learning Toolbox finden Sie in Hilfe-Center und File Exchange

Gefragt:

am 9 Dez. 2020

Kommentiert:

am 11 Dez. 2020

Community Treasure Hunt

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

Start Hunting!

Translated by