Toybrick

标题: 3588上的GPU可以用来做模型训练吗？ [打印本页]

作者: flyingroc 时间: 2022-5-18 11:18
标题: 3588上的GPU可以用来做模型训练吗？
如题，有无框架支撑？

作者: jefferyzhang 时间: 2022-5-18 11:32
本帖最后由 jefferyzhang 于 2022-5-18 15:13 编辑

当然可以。唯一能调用的GPU计算接口除了openGL外就是openCL和Vulkan。
而tensorflow和pytorch都可以支持cl训练和推理，默认官方预编译包只有cuda，你得自己去编译CL支持的版本

欢迎光临 Toybrick (https://t.rock-chips.com/)