Toybrick

标题: rk3399pro GPU使用 [打印本页]

作者: zjmmm    时间: 2020-7-24 16:13
标题: rk3399pro GPU使用
您好,我想请问下RK3399Pro 支持的cuda的使用吗?支持使用GPU跑pytorch或tensorflow的模型吗?


作者: leok    时间: 2020-7-27 11:38
不支持cuda 。 cuda是NV针对他们的GPU推出的并行计算架构。

3399pro上GPU进行推理,可以参考:https://github.com/OAID/Tengine
作者: flyingroc    时间: 2022-5-18 11:16
3399pro上GPU可以用来做模型训练用途吗?




欢迎光临 Toybrick (https://t.rock-chips.com/) Powered by Discuz! X3.3