潮声隔雨深 · 2020年02月13日

Mali GPU 支持tensorflow或者caffe等深度学习模型吗?

Mali GPU 支持tensorflow或者caffe等深度学习模型吗? 好像caffe2go和tensorflow lit可以部署到ARM,但不知道是否支持在GPU运行?

我希望把训练好的Tensorflow或者Caffe模型部署到ARM平台Mali-G71/72 GPU上运行,而不重新OpenCL编写代码,但没有看见相关可行的资料。网上信息显示tensorflow lit和caffe2Go可以部署到ARM,但不支持GPU?

1 个回答 得票排序 · 时间排序
一知半解 · 2020年02月13日

你好。ARM已经发布基于NEON的库来支持Tensorflow,基于Bifrost GPU(G71/72)的也会发布,到时候Mali GPU也可以基于这些库来支持Tensorflow和Caffe。

你的回答