0
Follow
5
View

Allocator ran out of memory - how to clear GPU memory from TensorFlow dataset?

czlaaaaaa 注册会员
2023-01-26 04:02

Try setting a hard limit on the total GPU memory as shown in here

import tensorflow as tf
gpus = tf.config.experimental.list_physical_devices('GPU')
tf.config.experimental.set_memory_growth(gpus[0], True)

About the Author

Question Info

Publish Time
2023-01-26 04:02
Update Time
2023-01-26 04:02

Related Question