Advertisement
Guest User

Untitled

a guest
Jun 15th, 2019
129
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.24 KB | None | 0 0
  1. config = tf.ConfigProto()
  2. config.gpu_options.allow_growth=True
  3. sess = tf.Session(config=config)
  4.  
  5. gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.333)
  6. sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement