#gpu #google-colaboratory #tensor #cudnn
Вопрос:
Я пытаюсь настроить детектор объектов SSD с помощью Tensorflow API. Я использую бесплатный графический процессор, предоставляемый Google cola. Но когда я пытаюсь обучить модель, это приводит меня к следующей ошибке.
Не удалось получить алгоритм свертки. Вероятно, это связано с тем, что cuDNN не удалось инициализировать, поэтому попробуйте посмотреть, было ли напечатано сообщение журнала предупреждений выше. [[узел ResNet101V1_FPN/модель/conv1_conv/Conv2D (определен в /local/lib/python3.7/dist-packages/object_detection/models/ssd_resnet_v1_fpn_keras_feature_extractor.py:224) ]] [Op:__вывод__фиктивный_вычисление_fn_21038].
Технические характеристики: Python 3.9 tensorflow 2.7.0 CUDA 11.2 cuDNN 8.1
Я попытался обновить cuDNN до 8.2, но получил ту же ошибку.
Кто-нибудь может предложить способ обновления cuDNN в Google Colab?