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]加载的运行时CuDNN库：7103（兼容版本7100），但源代码编译为7005（兼容版本7000）。如果使用二进制安装，请升级您的CuDNN库以使其匹配。如果从源构建，请确保在运行时加载的库与编译配置期间指定的兼容版本匹配。

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