利用GPU训练tensorflow模型时报错：Input Source operations connected to node average_pooling2d/AvgPool: ExpandD

1. 配置环境

Ubuntu：20.04.5

cuda:10.0

cudnn:7.6.0

tensorflow-gpu:1.14.0

2. 解决方法

(1)GPU设备选择

# 指定使用GPU的编号（GPU编号从0开始，一般为0）
CUDA_VISIBLE_DEVICES=0

（2）内存占有率

    config = tf.ConfigProto() 
    config.gpu_options.per_process_gpu_memory_fraction = 0.9 # 占用GPU90%的显存 
    config.log_device_placement=True
    sess = tf.Session(config=config) 

检查一下程序中是否多次设置内存占有率，比如在程序开头设置一次，在train（）方法中又设置一次。我的程序在修改后，就可以使用GPU训练了。