web-dev-qa-db-ja.com

モデルの重みを読み込んだ後の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バージョンが必要ですか？

tensorflowcudnnmagenta
10
user3776458

エラーが言うように、使用しているTensorflowバージョンはCuDNN 7.0.5用にコンパイルされていますが、システムにはCuDNN 7.1.3がインストールされています。

エラーが示唆するように、この問題を解決できます：

10
benjaminplanche

Envで：

ubuntu16.04   
cuda9.0   
cudnn7.0  
tensorflow 1.11.0  
python 3.5

私はテンソルフローでオブジェクト検出を訓練しようとします、私はこの問題に出会います：

2018-10-18 21:31:36.796017: E tensorflow/stream_executor/cuda/cuda_dnn.cc:343] Loaded runtime CuDNN library: 7.0.5 but source was compiled with: 7.2.1.  CuDNN library major and minor version needs to match or have higher minor version in case of CuDNN 7.0 or later version. If using a binary install, upgrade your CuDNN library.  If building from sources, make sure the library loaded at runtime is compatible with the version specified during compile configuration.
Segmentation fault (core dumped)

それは、テンソルフローのバージョンがより高いためです。

pip3 install --upgrade --force-reinstall tensorflow-gpu==1.9.0 --userを使用して問題を修正します。

6
zack

cudnnがコンパイルされているtensorflowのバージョンをインストールすることをお勧めします。

Sudo apt install libcudnn7-dev=7.0.5.15-1+cuda<x> libcudnn7=7.0.5.15-1+cuda<x>

<x>シンボルは、お持ちのcudaバージョンに置き換える必要があります。 cudaバージョン9.0の場合は、9.0に置き換えます。

後で自動的に更新されないaptのバージョンを後で凍結します。

Sudo apt-mark hold libcudnn7 libcudnn7-dev
0
Primoz