如何在使用带有TensorFlow后端的Keras时禁用cuDNN或强制它具有确定性行为?

时间:2018-01-08 19:47:51

标签: tensorflow keras cudnn

我在GPU上使用带有TensorFlow后端的Keras。如何禁用cuDNN或强制它具有确定性行为?

这里提到的Theano后端有一个明确的方法: https://github.com/keras-team/keras/issues/2479

但我找不到TensorFlow的类似方法。非常感谢任何帮助。

1 个答案:

答案 0 :(得分:0)

Tensorflow 目前不保证其所有操作的确定性行为。

您可以从这些帖子 1 & 2 中找到清晰的解释。