InternalEror :Dst tensor is not initialized

出现这个错误：InternalEror :Dst tensor is not initialized，一般是GPU的内存耗尽，需要在程序中加入一些语句

出现错误的程序：

        train_op = optimizer.minimize(loss_cross_entropy + l2_loss*weight_decay)
        saver=tf.train.Saver(max_to_keep=1)
        
###     begin training     ###    

#    0表示使用CPU，1则是GPU
    #config = tf.ConfigProto(device_count = {'GPU': 0})
    with tf.Session(config=config) as sess:
        sess.run(tf.global_variables_initializer())
        count_params()

需要在会话进行之前执行    config=tf.ConfigProto()      config.gpu_options.allow_growth = True语句，   

改正之后的程序：

        train_op = optimizer.minimize(loss_cross_entropy + l2_loss*weight_decay)
        saver=tf.train.Saver(max_to_keep=1)
        
###     begin training     ###    

    config=tf.ConfigProto()
    config.gpu_options.allow_growth = True
#    0表示使用CPU，1则是GPU
    #config = tf.ConfigProto(device_count = {'GPU': 0})
    with tf.Session(config=config) as sess:
        sess.run(tf.global_variables_initializer())
        count_params()