SPACES
SPACES copied to clipboard
预测时出现错误
苏神,请问下,我前面训练完了,预测的时候,出现“Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above.”错误,根据“config = tf.ConfigProto() config.gpu_options.allow_growth = True keras.backend.tensorflow_backend.set_session(tf.Session(config=config))” 修改了预测代码,又出现“tensorflow.python.framework.errors_impl.FailedPreconditionError”错误。难道就是因为我用的服务器tensorflow版本是1.13.1造成的吗?1.14版本我装不上,也不知道咋回事。代码小白,请教
装不上tf 1.14+那就用不了,bert4keras需要tf 1.14+(虽然我不知道你前面是怎么训练成功的)