set_memory_growth to all GPU devices alike

This commit is contained in:
Robert Sachunsky 2023-04-13 19:02:41 +02:00 committed by cneud
parent cb5ffaee14
commit 529f2c0e19

View file

@ -650,7 +650,8 @@ class Eynollah:
#session = tf.compat.v1.Session(config=tf.compat.v1.ConfigProto(gpu_options=gpu_options)) #session = tf.compat.v1.Session(config=tf.compat.v1.ConfigProto(gpu_options=gpu_options))
physical_devices = tf.config.list_physical_devices('GPU') physical_devices = tf.config.list_physical_devices('GPU')
try: try:
tf.config.experimental.set_memory_growth(physical_devices[0], True) for device in physical_devices:
tf.config.experimental.set_memory_growth(device, True)
except: except:
self.logger.warning("no GPU device available") self.logger.warning("no GPU device available")