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

pvsdev profile image

History of C and C++. Part one: emergence and standardization of C language. C with Classes becomes C++

Anastasiia Vorobeva -

tuttelikz profile image

Creating C++ DLL and using it in .NET

San Askaruly -

ambrus_pethes_a59563db94b profile image

Top 5 self-service BI solutions for Snowflake

Ambrus Pethes -

pentest_testing_corp profile image

Insecure Direct Object References (IDOR) in Laravel

Pentest Testing Corp -