时间戳

首页 > 解决方案 > tensorflow 2.0 中是否有 cudnnLSTM 或 cudNNGRU 替代方案

python - tensorflow 2.0 中是否有 cudnnLSTM 或 cudNNGRU 替代方案

问题描述

进货真的很快CuDNNGRU。TensorFlow 1.0但是当我转移到TensorFlow 2.0我无法找到CuDNNGRU。简单GRU是真的慢 TensorFlow 2.0。

有什么方法可以使用CuDNNGRU吗TensorFlow 2.0？

标签： pythontensorflowkerasrecurrent-neural-networktensorflow2.0

解决方案

可导入的实现已被弃用 - 相反，如果满足所有条件LSTM，GRU将默认为：CuDNNLSTMCuDNNGRU

  1. activation = 'tanh'
  2. recurrent_activation = 'sigmoid'
  3. recurrent_dropout = 0
  4. unroll = False
  5. use_bias = True
  6. 输入，如果被屏蔽，则严格右填充
  7. reset_after = True（仅限 GRU）

还要确保 TensorFlow 使用 GPU：

import tensorflow as tf
from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())
print('Default GPU Device: {}'.format(tf.test.gpu_device_name()))

更新：在 Colab 上运行时，TF 2.0.0 似乎存在问题，让 CuDNN 工作；试试吧!pip install tensorflow==2.1.0。

推荐阅读