深度学习

will not use cuDNN kernels since

2021-07-22  本文已影响0人  Qiangest

CUDNN运行是有一定的要求的:'

Keras API里有这样的说法:

Based on available runtime hardware and constraints, this layer will choose different implementations (cuDNN-based or pure-TensorFlow) to maximize the performance. If a GPU is available and all the arguments to the layer meet the requirement of the CuDNN kernel (see below for details), the layer will use a fast cuDNN implementation.

在这里插入图片描述
上一篇 下一篇

猜你喜欢

热点阅读