如何创建CNN与一个GPU在TensorFlow确定性操作?(How to create a CNN

2019-09-26 16:04发布

所以之后我知道有在TensorFlow这是不确定性的业务,看到这个问题: 如何获得相同的损耗值,每次训练CNN(MNIST数据集),与TensorFlow时间? , 我想知道:

我怎样才能建立一个卷积神经网络具有:

  • TensorFlow版本1.1.0
  • CUDA 8.0版本,V8.0.61
  • cuDNN 5.1.10
  • 在GPU上运行

其中只使用确定性的操作?

Answer 1:

你不能直到cuDNN每一个操作都不能完全确定。 此外,即使移动GPU上的每一个操作,如果它使用(很可能如此)SSE指令,你可以得到执行相同相同的(并且没有随机性)代码不止一次不同的结果。

你可以去接近你的目标的最好的事情,它设置seed为包含随机性(运一级的种子),每一个操作和设置整个图形(种子tf.set_random_seed(value) )。



文章来源: How to create a CNN with deterministic operations in TensorFlow on a GPU?