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()

        

    

你可能感兴趣的:(InternalEror :Dst tensor is not initialized)