Tensorflow可否使用共享GPU内存?

可以看下: 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