可以看下: config.proto
使用则是通过 ConfigProto.
gpu_options=tf.GPUOptions(per_process_gpu_memory_fraction=0.7)
config=tf.ConfigProto(gpu_options=gpu_options)
session = tf.Session(config=config, ...)
回帖人:dcmrlee
发表于 2018-6-9 15:11:55
可以看下: config.proto
使用则是通过 ConfigProto.
gpu_options=tf.GPUOptions(per_process_gpu_memory_fraction=0.7)
config=tf.ConfigProto(gpu_options=gpu_options)
session = tf.Session(config=config, ...)
回帖人:dcmrlee
发表于 2018-6-9 15:11:55