如何让TensorFlow使用我的GPU？

Question

我正在尝试运行一个名为 YOLO (darkflow)。在

我已经安装了cuda8.0，CudNN 5.1，tensorflow 1.0和tensorflow gpu（都是通过pip安装的）。在

我使用以下命令行测试YOLO：

sudo python3 ./flow --model cfg/yolo.cfg --load yolo.weights --demo dji_0004.MP4 --gpu 1.0

问题是：即使使用--gpu1.0，它也以~0.8fps的速度运行。在

在运行它时，我使用nvidia smi来监视我的GPU，这就是我得到的：

^{pr2}$

显然tensorflow没有在我的GPU上运行它。在

根据NVIDIA，GTX 650有3.0的计算能力，所以使用tensorflow gpu应该没有问题。在