是否可以在 keras\tensorflow 中关闭 CuDNN？答案

【问题标题】：Is it possible to switch off CuDNN in keras\tensorflow?是否可以在 keras\tensorflow 中关闭 CuDNN？
【发布时间】：2017-12-12 09:16:28
【问题描述】：

是否可以在 keras\tensorflow 中关闭 CuDNN？或者是否可以在没有 CuDNN 的情况下安装 keras\tensorflow（据我所知，anaconda CuDNN 是默认安装的）？

【问题讨论】：

Tensorflow 的仅 CPU 构建不需要 cuDNN。但是，如果您想选择 GPU 版本，则所有预构建的内容都是针对某些版本的 cuDNN 构建的。可能可以在编译时禁用它，但据我所知，GPU 构建说明要求您安装 cuDNN，所以我不太确定这是否可行。

【解决方案1】：

您可以像这样使特定层（或整个模型）仅使用 CPU：

with tf.device('/cpu:0'):
    #declare your model's layers here

【讨论】：