231
社区成员
发帖
与我相关
我的任务
分享
with tf.Graph().as_default():
# Start running operations on the Graph.能够在GPU上分配的最大内存
gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=args.gpu_memory_fraction)
sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options, log_device_placement=False))