我想在tensorflow lite中使用gpu操作方式,我可以为此研究

时间:2018-04-26 04:21:44

标签: gpu tensorflow-lite

首先,让我解释一下我必须做的事情。 我的发展环境是Tizen OS。可能你是无比的,无论如何这个操作系统使用基于Linux内核的redhat并瞄准移动,电视等。而我的目标设备是由exynos 5422和arm mali-t628组成。 我的主要工作是实现一些gpu库让tensorflow lite的操作可以使用该库。 我继续构建并安装tensorflow lite作为rpm包文件。 我正在谷歌搜索关于tensorflow和gpu。并得到一些关于cuda的无用信息。我没有看到我的情况的任何信息(tizen和mali gpu)。 我认为linux有像cpu或库这样的gpu指令..但我找不到它们。 你能建议搜索关键词或文件吗?

2 个答案:

答案 0 :(得分:0)

你可以访问nvidia的cuda工具包页面,在那里你可以找到documentationTraining按钮/选项。

还有CUDA programming guide我自己觉得这对CUDA非常有用和帮助。

我相信其中的一两个可以帮助你。

答案 1 :(得分:0)

CUDA适用于NVidia GPU。马里不是NVidia的,而是ARM的。因此,您不能在给定的硬件中使用CUDA。此外,如果您要使用CUDA,最好删除Tensorflow-lite,然后使用Tensorflow。

  1. 如果要使用CUDA,请获取具有受支持的NVidia GPU的硬件(例如,具有NVidia GPU的x64计算机)。请注意,您可以在Tizen中将Tensorflow-GPU和CUDA / CUDNN与x64 + NVidia GPU一起使用。您只需要注意nvidia GPU内核驱动程序版本和用户空间驱动程序版本。由于NVidia的GPU用户空间驱动程序和CUDA / CUDNN是静态构建的,因此其Linux驱动程序与Tizen兼容。 (我已经在Tizen中使用NVidia驱动程序111版测试了tensorflow-gpu,CUDA / CUDNN ...大概是在2017年冬天)
  2. 如果要在给定的硬件中使用Tizen / Tensorflow-lite,请不要使用CUDA。