Инициализация графического процессора для маски r-cnn + kaggle - PullRequest
0 голосов
/ 17 марта 2020

Я установил несколько унаследованных пакетов с тензорным потоком, чтобы можно было обучить эту более старую модель MASK R-CNN из материального порта. Tensorflow-GPU == 1.13.1

Я не уверен, нужно ли мне указать docker изображение? Я знаю, что TF2.1 использует CudNN 7.6 - tenorflow-gpu 1.13.1 говорит, что хочет CudNN 7.4 - или если мне не хватает чего-то очень простого.
https://www.tensorflow.org/install/source#tested_build_configurations

Я вручную уменьшил количество шагов / эпох и количество эпох, чтобы на ядре работало около 10 минут.

Как мне заставить этот графический процессор Kaggle работать на этом ядре ?:

https://www.kaggle.com/stargarden/m-r-cnn-matterport-1

...