diff --git a/modules/deeplearning/cloudheight.py b/modules/deeplearning/cloudheight.py index ae3bad25935c4bc5568fc71d9afefdfd51cc925d..86724250e4ab46b9caa206317e2fa22a67bbedba 100644 --- a/modules/deeplearning/cloudheight.py +++ b/modules/deeplearning/cloudheight.py @@ -250,6 +250,7 @@ class CloudHeightNN: tf.debugging.set_log_device_placement(LOG_DEVICE_PLACEMENT) gpus = tf.config.experimental.list_physical_devices('GPU') + print(gpus) if gpus: try: # Currently, memory growth needs to be the same across GPUs