Я пытаюсь использовать версию tenorflow gpu для обучения и тестирования моей модели глубокого обучения.Но тут возникает проблема.Когда я тренирую свою модель в одном файле Python, дела идут хорошо.Tensorflow-GPU можно использовать правильно.Затем я сохраняю свою модель как предварительно обученный в формате grapg.pb и пытаюсь повторно использовать ее в другом файле Python.
Затем я получил следующие сообщения об ошибках.
E tensorflow/stream_executor/cuda/cuda_dnn.cc:363] Loaded runtime CuDNN
library: 7.1.4 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.
Я проверил свою версию cudnn, фактически это версия 7.4.2.Я также проверил настройки моего окружения /cuda/v9.0/bin, cuda/v9.0/lib/x64, /cuda/v9.0/include.
Так почему это происходит?И как я могу решить это?
-
cuda: v9.0 cudnn: 7.4.2 (я думаю, я копирую эти файлы cudnn вручную) windows 10 python: 3.5