diff --git a/modules/deeplearning/icing_cnn.py b/modules/deeplearning/icing_cnn.py index 1b5c34b8c93f3926f2535997d662827644874f86..3b582a5809f97c9afca13296ee8f547e6f7eef3a 100644 --- a/modules/deeplearning/icing_cnn.py +++ b/modules/deeplearning/icing_cnn.py @@ -217,17 +217,17 @@ class IcingIntensityNN: tf.debugging.set_log_device_placement(LOG_DEVICE_PLACEMENT) - gpus = tf.config.experimental.list_physical_devices('GPU') - if gpus: - try: - # Currently, memory growth needs to be the same across GPUs - for gpu in gpus: - tf.config.experimental.set_memory_growth(gpu, True) - logical_gpus = tf.config.experimental.list_logical_devices('GPU') - print(len(gpus), "Physical GPUs,", len(logical_gpus), "Logical GPUs") - except RuntimeError as e: - # Memory growth must be set before GPUs have been initialized - print(e) + # gpus = tf.config.experimental.list_physical_devices('GPU') + # if gpus: + # try: + # # Currently, memory growth needs to be the same across GPUs + # for gpu in gpus: + # tf.config.experimental.set_memory_growth(gpu, True) + # logical_gpus = tf.config.experimental.list_logical_devices('GPU') + # print(len(gpus), "Physical GPUs,", len(logical_gpus), "Logical GPUs") + # except RuntimeError as e: + # # Memory growth must be set before GPUs have been initialized + # print(e) def get_in_mem_data_batch(self, idxs, is_training):