Обнаружение объекта Tensorflow Api UnknownError (см. Выше для отслеживания): не удалось получить алгоритм свертки

Question

Когда я пытался тренировать модель с

python3 model_main.py —logtostderr —train_dir=training/ —pipeline_config_path=training/ssd_mobilenet_v1_pets.config    

Я получил следующую ошибку

UnknownError (see above for traceback): Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try  looking to see if a warning log message was printed above.
[[node FeatureExtractor/MobilenetV1/MobilenetV1/Conv2d_0/Conv2D (defined at /home/paperspace/Desktop/models/research/slim/nets/mobilenet_v1.py:266) ]]
[[node Loss/unstack (defined at /home/paperspace/Desktop/models/research/object_detection/meta_architectures/ssd_meta_arch.py:1073) ]]

Ubuntu 18.10

Моя версия Tensorflow - 1.13.1

Cuda

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2018 NVIDIA Corporation
Built on Sat_Aug_25_21:08:01_CDT_2018
Cuda compilation tools, release 10.0, V10.0.130

Понижение до 1.8.1 Tensorflow не является решением Что случилось? Вероятно, это может привести к конфликту зависимостей

Answer 1

Попробуйте добавить эти строки кода сразу после импорта тензорного потока

from tensorflow.compat.v1 import ConfigProto
from tensorflow.compat.v1 import InteractiveSession

config = ConfigProto()
config.gpu_options.allow_growth = True
session = InteractiveSession(config=config)

Answer 2

Ну, я не знаю.Поскольку у меня есть чистая версия Ubuntu (без предустановочных библиотек), я управлял всеми зависимостями вручную. Понижение до 1.8.1 может быть решением в вашем случае, но не в моем.

Среда, которая работала для меня,

cudnn 7.5.1.10-1+cuda10.0
Tensorflow - 1.13.1