tensorflow - 使用 CUDA 运行的 TensorFlow 对象检测 API

Question

当我尝试在我的自定义数据集上运行 tensorflow 对象检测 API 时，我收到以下错误消息。该进程刚刚被杀死。它在某种程度上与计算机内存有关吗？也许是 CUDA 版本或 tensorflow 版本？我有 tensorflow-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

标签： tensorflowmachine-learningdeep-learningobject-detection