2023年12月6日
自有AI芯片接入AI框架Pytorch的方案
摘要： 现在AI框架主要用Pytorch，包括一些常用的库对Pytorch支持都较好，国产AI芯片很多也对AI框架Pytorch进行了支持 一、华为昇腾npu 能够跟上Pytorch的更新，直接和Pytorch兼容，而且有较多人来维护，代码风格不错，之前是通过注入Pytorch预留的的xla搞的接入，Pyt 阅读全文
posted @ 2023-12-06 12:31 暴力都不会的蒟蒻 阅读(1128) 评论(0) 推荐(0) 编辑