Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- config = tf.ConfigProto()
- config.gpu_options.allow_growth=True
- sess = tf.Session(config=config)
- gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.333)
- sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement