-
Notifications
You must be signed in to change notification settings - Fork 0
Home
Iulian Mocanu edited this page Jan 27, 2021
·
3 revisions
Python code for GPU optimization:*
config = tf.compat.v1.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.9
tf.compat.v1.keras.backend.set_session(tf.compat.v1.Session(config=config))
Good to read :