From 6d324f55e65209d22f049b57e15a8951d9385a90 Mon Sep 17 00:00:00 2001
From: Hykilpikonna <me@hydev.org>
Date: Tue, 10 May 2022 00:00:01 -0400
Subject: [PATCH] [O] Limit GPU usage

---
 experiment/ina_main.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/experiment/ina_main.py b/experiment/ina_main.py
index 338b133..5032e82 100644
--- a/experiment/ina_main.py
+++ b/experiment/ina_main.py
@@ -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()