相关疑难解决方法(0)

我正在尝试加载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版本？