如何在GPU而不是CPU上使用Tensorflow?

时间:2018-06-19 20:05:12

标签: tensorflow gpu object-detection-api

我目前正在关注Tensorflow对象检测API教程(details linked here)。我已经成功地开始训练我的模型,但是训练没有使用我的机器GPU。我怀疑是这种情况,因为它运行非常缓慢,并且在执行过程中我还收到以下警告:

“您的CPU支持该TensorFlow二进制文件未编译为使用的指令:SSE4.1 SSE4.2”

如何强制训练过程使用我的GPU?我尝试卸载tensorflow并重新安装不同的版本,但是它不起作用。 Tensorflow-gpu肯定已安装。有任何想法吗?我正在使用Ubuntu 14.04。

0 个答案:

没有答案