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

Question

我正在尝试加载 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] Loaded runtime CuDNN library: 7103 (compatibility version 7100) but source was compiled with 7005 (compatibility version 7000). If using a binary install, upgrade your CuDNN library to match. If building from sources, make sure the library loaded at runtime matches a compatible version specified during compile configuration.

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

Answer 1

正如错误所说，您使用的 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)。