ARM技术论坛
直播中

juju宇哥

8年用户 1406经验值
擅长:479809
私信 关注
[问答]

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)

张红

2022-9-16 14:13:26
你好。ARM已经发布基于NEON的库来支持Tensorflow,基于Bifrost GPU(G71/72)的也会发布,到时候Mali GPU也可以基于这些库来支持Tensorflow和Caffe。
举报

更多回帖

发帖
×
20
完善资料,
赚取积分