API обнаружения объектов Tensorflow, работающий с CUDA - PullRequest
0 голосов
/ 07 января 2020

Когда я пытаюсь запустить API обнаружения объектов tenorflow в моем пользовательском наборе данных, я получаю сообщение об ошибке ниже. Процесс просто убит. Это как-то связано с памятью компьютера? Может быть, версия CUDA или версия tenorflow? У меня есть tenorflow-gpu версии 1.13.1 и CUDA версии 10.0.

Любая помощь будет оценена!

I tensorflow/stream_executor/dso_loader.cc:152] successfully opened CUDA library libcublas.so.10.0 locally
INFO:tensorflow:global_step/sec: 0
INFO:tensorflow:global_step/sec: 0
Killed
...