现在AI框架主要用Pytorch,包括一些常用的库对Pytorch支持都较好
一、华为昇腾npu
能够跟上Pytorch的更新,直接和Pytorch兼容,而且有较多人来维护,代码风格不错,之前是通过注入Pytorch预留的的xla
搞的接入,现在被官方接收了。非常推荐,笔者用他的框架实现了自有GPGPU芯片手写数字识别的训练
二、寒武纪mlu
这个版本很老,他们是通过docker为客户提供运行环境的,pytorch_patches文件夹也表明他们需要修改Pytorch源代码
三、谷歌tpu
Pytorch 给他留了接口,甚至aten中也有为了支持他的某些feature的源码
四、OpenCL 实现
一个老哥在孜孜不倦得推动OpenCL在Pytorch的适配,他还一直在维护,可惜OpenCL都要无了(
五、摩尔线程 MTT
不知道他开源在哪里了,github团队是空的,可能合作商能从他官网拿到?他说自己开源了, 2023夏季发布会
标签:框架,AI,OpenCL,接入,芯片,Pytorch From: https://www.cnblogs.com/BobHuang/p/17879241.html