From f0d81e90a4ec765992b92336e343832f31d15ee0 Mon Sep 17 00:00:00 2001 From: tomrink <rink@ssec.wisc.edu> Date: Mon, 6 Mar 2023 12:59:07 -0600 Subject: [PATCH] snapshot... --- modules/deeplearning/cnn_cld_frac_mod_res.py | 22 ++++++++++---------- 1 file changed, 11 insertions(+), 11 deletions(-) diff --git a/modules/deeplearning/cnn_cld_frac_mod_res.py b/modules/deeplearning/cnn_cld_frac_mod_res.py index df363c4b..df09941a 100644 --- a/modules/deeplearning/cnn_cld_frac_mod_res.py +++ b/modules/deeplearning/cnn_cld_frac_mod_res.py @@ -110,17 +110,17 @@ if DO_ESPCN: y_128 = slice(2, 130) -gpus = tf.config.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.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.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.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 build_residual_conv2d_block(conv, num_filters, block_name, activation=tf.nn.relu, padding='SAME', -- GitLab