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

Question

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

Answer 1

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

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