Маска RCNN использует процессор вместо графического процессора

Question

Я использую библиотеку Mask RCNN, основанную на tenserflow, и я не могу заставить ее работать на моем GPU (1080TI).Время вывода составляет 4-5 секунд, в течение которого я вижу всплеск использования моего процессора, но не моего процессора.Любые возможные исправления для этого?

Answer 1

Это либо потому, что для GPU_COUNT установлено значение 0 в config.py, либо у вас не установлено tensorflow-gpu (что требуется для запуска tenorflow на GPU)