我有一台带有Windows 10和1080ti gpu的电脑(i5,16G RAM)。我安装了TF 1.4,python anaconda 3.6,cuda 8.0和cuDNN v6.0。
我正在根据dtran's训练ssd mobilenet对象检测器。培训运行速度低于预期:
INFO:tensorflow:global step 14463:loss = 1.1131(2.125 sec / step)
INFO:tensorflow:global step 14464:loss = 1.1103(2.094 sec / step)
INFO:tensorflow:全球步骤14465:损失= 0.8764(2.141秒/步)
INFO:tensorflow:global step 14466:loss = 0.9378(2.391 sec / step)
我如何判断一切是否正常运行,这是预期的性能还是有问题?是否有TF的基准工具你可以下载,运行并将结果与预期结果进行比较吗?迁移到ubuntu会改善结果吗?
答案 0 :(得分:-1)
我希望你的问题能在这个时候解决,如果不是...... 我在第一次设置并在Windows中运行我的程序时遇到了类似的问题,后来我发现windows中的tensorflow gpu版本需要python 3.5而不是3.6。我猜你的程序使用的是CPU而不是GPU。如果您使用的是像Pycharm这样的IDE,请选择tensorflow GPU env作为python解释器。当你运行你的python程序时,它应该显示一些像“/ gpu:0”:你的机器的GPU。