tensorflow - 加载模型权重后CuDNN库兼容性错误

我正在尝试加载NSynth权重，并且我正在使用TF版本1.7.0

from magenta.models.nsynth import utils
from magenta.models.nsynth.wavenet import fastgen

def wavenet_encode(file_path):

 # Load the model weights.
 checkpoint_path = './wavenet-ckpt/model.ckpt-200000'

 # Load and downsample the audio.
 neural_sample_rate = 16000
 audio = utils.load_audio(file_path,
                          sample_length=400000,
                          sr=neural_sample_rate)

 encoding = fastgen.encode(audio, checkpoint_path, len(audio))

 # Reshape to a single sound.
 return encoding.reshape((-1, 16))

# An array of n * 16 frames.
wavenet_z_data = wavenet_encode(file_path)

我收到以下错误：


tensorflow / stream_executor / cuda / cuda_dnn.cc：396]加载的运行时CuDNN
库：7103（兼容版本7100），但源已编译
与7005（兼容版本7000）。如果使用二进制安装，
升级您的CuDNN库以使其匹配。如果从源头建造，
确保在运行时加载的库与兼容版本匹配
在编译配置期间指定。


我该怎么办，应该安装哪个版本的tf，以及我到底需要哪个CUDA版本？

最佳答案

如错误所述，您使用的Tensorflow版本是针对CuDNN 7.0.5编译的，而系统已安装CuDNN 7.1.3。

正如错误所暗示的，您可以解决此问题：


通过安装CuDNN 7.0.5（按照此处的说明：https://developer.nvidia.com/cudnn）进行；
或者通过自己为系统编译Tensorflow（按照此处的说明：https://www.tensorflow.org/install/install_sources）。

关于tensorflow - 加载模型权重后CuDNN库兼容性错误，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/49960132/