Toybrick

rk3399pro GPU使用

zjmmm

中级会员

积分
304
发表于 2020-7-24 16:13:01    查看: 7845|回复: 2 | [复制链接]    打印 | 显示全部楼层
您好,我想请问下RK3399Pro 支持的cuda的使用吗?支持使用GPU跑pytorch或tensorflow的模型吗?

回复

使用道具 举报

leok

版主

积分
894
发表于 2020-7-27 11:38:46 | 显示全部楼层
不支持cuda 。 cuda是NV针对他们的GPU推出的并行计算架构。

3399pro上GPU进行推理,可以参考:https://github.com/OAID/Tengine
回复

使用道具 举报

flyingroc

注册会员

积分
114
发表于 2022-5-18 11:16:48 | 显示全部楼层
3399pro上GPU可以用来做模型训练用途吗?
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

产品中心 购买渠道 开源社区 Wiki教程 资料下载 关于Toybrick


快速回复 返回顶部 返回列表