tensorflow-densenet icon indicating copy to clipboard operation
tensorflow-densenet copied to clipboard

Published 20 hours ago • verified publisher icon pudae

how to set the usage of gpu

Open lxztju opened this issue 6 years ago • 0 comments

In tensorflow, we could use these commends to set the usage of gpu config = tf.ConfigProto() config.gpu_options.per_process_gpu_memory_fraction = 0.9 # 占用GPU90%的显存 session = tf.Session(config=config) If I use the package of slim, how to set the usage of gpu?

lxztju avatar Jan 04 '19 02:01 lxztju