Tensorflow对象检测训练性能

时间:2018-04-10 12:47:20

标签: performance tensorflow gpu object-detection

我正在使用针对张量流的对象检测API在我自己的自定义类上训练ssd_mobilenet_v1_coco网络。

我使用CPU(i7-6700)和GPU(NVIDIA Quadro K620)进行训练:

Processor   Batch size  sec/step    sec/image
K620        1           0,45        0,450
K620        10          2,22        0,222
i7-6700     1           0,66        0,660
i7-6700     24          9,3         0,388

然而,GPU仅比CPU快约70%。 我预计GPU会更快。 这种性能是否适合我的硬件或是否有问题?

1 个答案:

答案 0 :(得分:-1)

也许您可以尝试Tensorflow SERVER