delete tensorflow memory limit. Just use memory growths setting

tandav · Mar 31, 2023 · d414904 · d414904
1 parent 99873e7
commit d414904
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/pitch_detectors/algorithms/base.py b/pitch_detectors/algorithms/base.py
@@ -45,11 +45,9 @@ def __init__(self) -> None:
     def set_memory_limit(self) -> None:
         if TensorflowGPU.memory_limit_initialized:
             return
-        memory_limit = int(os.environ.get('PITCH_DETECTORS_GPU_MEMORY_LIMIT', 8192))
         gpus = self.gpus
         for gpu in gpus:
             self.tf.config.experimental.set_memory_growth(gpu, True)
-            self.tf.config.experimental.set_virtual_device_configuration(gpu, [self.tf.config.experimental.VirtualDeviceConfiguration(memory_limit=memory_limit)])
         TensorflowGPU.memory_limit_initialized = True
 
     @property