DEV Community

Cover image for Preventing Keras to allocate unnecessary GPU memory
Talles L
Talles L

Posted on

Preventing Keras to allocate unnecessary GPU memory

#keras 
gpus = tf.config.experimental.list_physical_devices('GPU')

for gpu in gpus:
  tf.config.experimental.set_memory_growth(gpu, True)
Enter fullscreen mode Exit fullscreen mode

Top comments (0)

Read next

xun_wang_6384a403f9817c2 profile image

Cross-Application Collaboration of Custom Data Types in HarmonyOS Next: Implementing Enterprise-Level Document Management

SameX -

peytonstrahan profile image

JavaScript Variable Keywords (Var, Let, and Const)

Peyton Strahan -

xun_wang_6384a403f9817c2 profile image

HarmonyOS Next自定义数据类型的跨应用协作：实现企业级文档管理

SameX -

annavi11arrea1 profile image

PSU Drama: Unwitty Resolve XD

Anna Villarreal -