From 67f6ecbbb9c2be2fc2c51467d495f3754887881f Mon Sep 17 00:00:00 2001 From: Malte Grosse Date: Thu, 11 Apr 2024 16:44:03 +0900 Subject: [PATCH] try without limit --- run.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/run.py b/run.py index 5d3aaa8..83de8ef 100644 --- a/run.py +++ b/run.py @@ -9,12 +9,12 @@ from keras import backend as K K.clear_session() gpus = tf.config.experimental.list_physical_devices('GPU') -if gpus: - try: - tf.config.experimental.set_virtual_device_configuration(gpus[0], [tf.config.experimental.VirtualDeviceConfiguration(memory_limit=6024)]) - except RuntimeError as e: - print(e) - os.exit(1) +# if gpus: +# try: +# tf.config.experimental.set_virtual_device_configuration(gpus[0], [tf.config.experimental.VirtualDeviceConfiguration(memory_limit=6024)]) +# except RuntimeError as e: +# print(e) +# os.exit(1) print(tf.config.experimental.list_physical_devices())