0

我有 TensorFlow 1.14、Keras 2.3.1、CUDA 10、windows。我在 GPU 上运行一个代码,效果很好,如果我通过使用with tf.device("/cpu:0"):命令在 CPU 上运行另一个代码,它也可以正常工作。但是如果我在 GPU 上运行一个代码并想同时在 CPU 上运行另一个代码,则该with tf.device("/cpu:0"):命令不起作用并产生错误:无法获取卷积算法。这可能是因为 cuDNN 未能初始化。我想知道我是否在使用 CPU,为什么 TF 还在寻找 cudNN?以及如何避免这种情况?

4

0 回答 0