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