你怎么知道你';是否在TensorFlow 2.0中重新使用LSTM或GRU的CuDNN版本?
TensorFlow 2.0的一个很好的特性是:现在自动决定是否使用CuDNN实现(层也是如此)。但它们的使用不仅仅取决于GPU;还需要设置其他参数(例如,你怎么知道你';是否在TensorFlow 2.0中重新使用LSTM或GRU的CuDNN版本?,tensorflow,keras,lstm,recurrent-neural-network,Tensorflow,Keras,Lstm,Recurrent Neural Network,TensorFlow 2.0的一个很好的特性是:现在自动决定是否使用CuDNN实现(层也是如此)。但它们的使用不仅仅取决于GPU;还需要设置其他参数(例如,unroll=False) 我使用的是tensorflow gpu和gpu(不是很强大),我的RNN速度很慢。我想确定我正在运行CuDNN实现。日志里有什么地方可以看吗 我也用过,对我来说也一样慢。我是否可以确定这是在使用CuDNN实现,或者在某些情况下它可能会退回到其他方式?如果您使用的是tf.compat.v1.keras.layers.
unroll=False
)
我使用的是tensorflow gpu和gpu(不是很强大),我的RNN速度很慢。我想确定我正在运行CuDNN实现。日志里有什么地方可以看吗
我也用过,对我来说也一样慢。我是否可以确定这是在使用CuDNN实现,或者在某些情况下它可能会退回到其他方式?如果您使用的是
tf.compat.v1.keras.layers.CuDNNLSTM
,那么肯定是在使用CuDNN实现。那里没有退路