Как загрузить модель тензорного потока в gpu в проекте Java - PullRequest
0 голосов
/ 28 июня 2019

Я пытаюсь запустить проект Java, который загружает модель тензорного потока для обнаружения объектов. Как запустить этот проект в gpu.

В моей системе установлена ​​графическая карта NVIDIAЯ уже запустил это в процессоре. Теперь я хочу переключиться на GPU.

Для переключения на графический процессор, следующие конфигурации былидобавлено. Но не удалось переключиться на графический процессор.

ConfigProto configproto = ConfigProto.newBuilder()
            .setAllowSoftPlacement(true)                 
            .setGpuOptions(GPUOptions.newBuilder()
            .setPerProcessGpuMemoryFraction(0.01).build())
            .build();

SavedModelBundle model = SavedModelBundle.loader(modeldir).withTags("serve").withConfigProto(configproto.toByteArray()).load()

Я ожидаю, что модель будет работать в графическом процессоре. Но полученный результат выглядит следующим образом

2019-06-28 14: 50: 25.652043: I tenorflow / cc / save_model / reader.cc: 31] Чтение SavedModel из: / save_model

2019-06-28 14: 50: 25.686901: Itenorflow / cc / save_model / reader.cc: 54] Чтение мета-графа с тегами {serve}

2019-06-28 14: 50: 25.699563: I tenorflow / core / platform / cpu_feature_guard.cc: 141]Ваш ЦП поддерживает инструкции, которые этот двоичный файл TensorFlow не был скомпилирован для использования: SSE4.1 SSE4.2 AVX AVX2 FMA

2019-06-28 14: 50: 25.704712: I тензор потока / ядро ​​/ платформа / profile_utils / cpu_utils.cc: 94] Частота ЦП: 2712000000 Гц

2019-06-28 14:50: 25.704847: I tensflow / compiler / xla / service / service.cc: 150] Служба XLA 0x7f6cf036be50, выполняющая вычисления на платформе Host.Устройства:

2019-06-28 14: 50: 25.704862: I tenorflow / compiler / xla / service / service.cc: 158] Устройство StreamExecutor (0):,

2019-06-28 14: 50: 25.759224: I tenorflow / cc / save_model / loader.cc: 182] Восстановление пакета SavedModel.

2019-06-28 14: 50: 25.759280: I tenorflow / cc / save_model / loader.cc: 192] Указанный SavedModel не имеет переменных;контрольно-пропускные пункты не были восстановлены.Файл не существует: /saved_model/variables/variables.index

2019-06-28 14: 50: 25.759291: I tenorflow / cc / save_model / loader.cc: 285] Загрузка SavedModel для тегов {serve};Статус: успех.Прошло 107259 микросекунд.

...