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

trinityyi profile image

REST vs. GraphQL: Choosing the Right API for Your Project

Isabelle M. -

philip_zhang_854092d88473 profile image

Secrets to Designing a Great API

Philip -

jeet_dhandha profile image

Building and Comparing Table Data Made Easy with @libs-jd/table-data-kit

Jeet Dhandha -

gutmaster profile image

Validatorian

Chevy Vall -