[O] Limit GPU usage

2022-05-10 00:00:01 -04:00
parent c37788c86a
commit 6d324f55e6
1 changed files with 5 additions and 0 deletions
@@ -16,8 +16,13 @@ from PIL import Image
 from inaSpeechSegmenter import *
 from matplotlib.axes import Axes
 from matplotlib.figure import Figure
+import tensorflow as tf


+gpus = tf.config.experimental.list_physical_devices('GPU')
+for gpu in gpus:
+    tf.config.experimental.set_memory_growth(gpu, True)
+
 seg = Segmenter()