У меня проблема с запуском приложения c ++ tenorflow сразу после перезагрузки Ubuntu.
обычно я строю приложение от bazel build
bazel build -c opt --local_resources 3072,4.0,1.0 --verbose_failures --config=monolithic //tensorflow/app/...
после сборки, когда я выполняю
bazel-bin/tensorflow/app/run
все работает отлично, но, когда я перезагружаю Ubuntu и запускаю сначала bazel-bin, у меня появляется спам об этих ошибках.
May 11 12:29:05 tegra-ubuntu start.sh[800]: 2018-05-11 12:29:04.622723: E tensorflow/stream_executor/cuda/cuda_driver.cc:967] failed to alloc 2304 bytes on host: CUDA_ERROR_UNKNOWN
May 11 12:29:05 tegra-ubuntu start.sh[800]: 2018-05-11 12:29:04.622849: E tensorflow/stream_executor/cuda/cuda_driver.cc:967] failed to alloc 2304 bytes on host: CUDA_ERROR_UNKNOWN
Полагаю, это проблема в загруженных пакетах. Когда сборка bazel выполняется до запуска bazel, загрузка пакетов выполняется правильно.
Есть способ загрузить только отсутствующие пакеты без предварительной сборки? Мне нужно запустить приложение сразу после запуска, но мой процесс сборки слишком длинный, чтобы запустить приложение до запуска приложения
Решение:
SessionOptions opts;
opts.config.mutable_gpu_options()->set_allow_growth(true);
session->reset(tensorflow::NewSession(opts));