首页 > 解决方案 > 我可以期望 GPU 上的 Tensorflow 比 CPU 带来多少性能提升?

问题描述

我在 Linux Mint 18 上安装了 tensorflow-gpu。我的显卡是 GT 740m。CUDA的deviceQueryandbandwidthTest和 cudnn 脚本的 MNISTsample 通过(在此处此处引用)。

Tensorflow 确实使用了 GPU(例如,遵循这些指令可以工作,并且在运行程序时 GPU 的内存和处理利用率会增加),但性能相当……平庸。

例如,运行此站点上显示的脚本,GPU 的速度仅为 CPU 的两倍左右。正如网站上所说,这当然是一个不错的改进,但不是“非常非常快”。另一个例子:使用带有 Keras 的 vgg16 对 100 张图像进行分类,每张大约 300x200 像素大约需要 30 秒。

我可以做些什么来提高性能,或者我不能期待更好的事情吗?

搜索查询:慢,

标签: tensorflow

解决方案


推荐阅读