Tensorflow实战之多GPU并行程序的实现

import tensorflow as tf
from tensorflow.examples.tutorials.mnist import input_data
import mnist_inference
import numpy as np
#模型保存路径
model_save_path = 'C:\\Users\\Administrator\\Desktop\\LZC\\model_save_path\\'
summary_save_path = 'C:\\Users\\Administrator\\Desktop\\LZC\\summary_save_path\\'
#数据路径
data_path = 'C:\\Users\\Administrator\\Desktop\\LZC\\path\\mnist.tfrecords'
#训练次数
train_step = 10000


#将mnist数据转为tfrecord
def  mnist_to_TFRecord(mnist):
    images = mnist.train.images
    labels = mnist.train.labels
    num = mnist.train.num_examples
    writer = tf.python_io.TFRecordWriter(data_path)
    for index in range(num):
        image = images[index].tostring()
        label = labels[index].tostring()
        examples = tf.train.Example(features=tf.train.Features(feature={
            'image': tf.train.Feature(bytes_list=tf.train.BytesList(value=[image])),