mirror of
https://github.com/qurator-spk/eynollah.git
synced 2025-06-09 04:09:54 +02:00
set_memory_growth to all GPU devices alike
This commit is contained in:
parent
cb5ffaee14
commit
529f2c0e19
1 changed files with 2 additions and 1 deletions
|
@ -650,7 +650,8 @@ class Eynollah:
|
|||
#session = tf.compat.v1.Session(config=tf.compat.v1.ConfigProto(gpu_options=gpu_options))
|
||||
physical_devices = tf.config.list_physical_devices('GPU')
|
||||
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:
|
||||
self.logger.warning("no GPU device available")
|
||||
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue