Вывод тензорного потока очень медленный при первом запуске в докере - PullRequest
0 голосов
/ 01 ноября 2018

Я создал образ докера для сервера, который делает вывод с тензорным потоком. Я установил tenorflow-gpu с pip в образе докера. Он отлично работает для моей машины с Titan X GPUS. Но когда я запустил док-контейнер на другой машине с 1080 ti gpus. Первый запуск становится невероятно медленным, занимает около 90 секунд, обычно это занимает 7 секунд при первом запуске и 1 секунду в следующих. Я попытался установить tf_cudnn_use_autotune в 0, а также смонтировать папку для сохранения кэша cuda. Но это на самом деле не решает проблему. У кого-нибудь есть предложения?

1 Ответ

0 голосов
/ 29 июля 2019

Вот ссылка . Я нахожу это.

После однократного запуска TensorFlow скомпилированные ядра кэшируются CUDA. При использовании Docker-контейнера данные не кэшируются, и каждый раз при запуске TensorFlow .

выплачивается штраф.
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...