AI Pool
Log in
Discussions>Prevent by allocating the whole gpu memory by tensorflow>

Prevent by allocating the whole gpu memory by tensorflow

How to prevent using whole memory of gpu by tensorflow 1.13?

tensorflowgpumemory
2 votesJW297.00
1 Answers
JO266.00
2

Tensorflow has some confirguarionts for every Session. You can just change the config for the session by using

config = tf.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.5
tf.Session(config=config)
Reply
Couldn't find what you were looking for?and we will find an expert to answer.
How helpful was this page?