tensorflow.python.framework.errors_impl.InvalidArgumentError: No OpKernel was registered to support Op 'MaxBytesInUse' used by node GPUMemoryTracker/MaxBytesInUse about attentionocr

[zhang0jhon]

程序调用GPU了么？

[wyc2015fq]

@zhang0jhon 我按照你的设置跑的，测试是正常的，训练就出现这个问题

[zhang0jhon]

config.py 里的GPU设置改了么？

[wyc2015fq]

@zhang0jhon 我的卡是P4的，有四个显卡。设置成gpus = [0, 1, 2, 3]，请问cuda 必须是10的吗？

[zhang0jhon]

可以尝试把train.py里面的GPUMemoryTracker()等Tracker的callbacks注释掉，我默认使用的是CUDA10 CUDNN7

[deeplearningshare]

@zhang0jhon 我的卡是P4的，有四个显卡。设置成gpus = [0, 1, 2, 3]，请问cuda 必须是10的吗？
cuda9也可以，但是需要将tensorflow版本降低，同时更改对应的一些接口就可以了，我用的是tensorflow 1.8.0，cuda9
接口更改比如：
with tf.compat.v1.variable_scope(name, reuse=reuse) as scope:
改成：
with tf.variable_scope(name, reuse=reuse) as scope:
类似这些
其中model.py中的
h = tf.nn.dropout(h, rate = 1-keep_prob)
c = tf.nn.dropout(c, rate = 1-keep_prob)
改成
h = tf.nn.dropout(h, keep_prob = keep_prob)
c = tf.nn.dropout(c, keep_prob = keep_prob)