Tensorflow 转换在CUDNNLSTM中训练的模型，以便训练的模型也可以在CPU上运行
我有一个受过CUDNNLSTM训练的模型，如何在CPU中使用它？我们如何将CUDNNLSTM变量导出到CPU基权重，以便训练后的模型也可以在CPU上运行。一种始终有效的安全方法是将变量保存到具有适当格式（如hdf5）的文件中，然后再次手动加载它们。这样，您就可以完全控制正在发生的事情，而不依赖于您正在使用的系统。您甚至可以在TensorFlow中训练一个模型，保存变量并加载它们，以便与其他库（如PyTorch）一起使用（假设您在那里定义了完全相同的模型）。当我将模型冻结到pb文件中并使用该文件运行时，我得到以下错误：未找到：Op类型未在上运行的二进制文件中注册“cudnrnn”。确保在该进程中运行的二进制文件中注册了Op和内核Tensorflow 转换在CUDNNLSTM中训练的模型，以便训练的模型也可以在CPU上运行,tensorflow,cudnn,Tensorflow,Cudnn,我有一个受过CUDNNLSTM训练的模型，如何在CPU中使用它？我们如何将CUDNNLSTM变量导出到CPU基权重，以便训练后的模型也可以在CPU上运行。一种始终有效的安全方法是将变量保存到具有适当格式（如hdf5）的文件中，然后再次手动加载它们。这样，您就可以完全控制正在发生的事情，而不依赖于您正在使用的系统。您甚至可以在TensorFlow中训练一个模型，保存变量并加载它们，以便与其他库（如PyTorch）一起使用（假设您在那里定义了完全相同的模型）。当我将模型冻结到pb文件中并使用该文件运