Pytorch 2.1版本已同步支持华为昇腾NPU

来源:鞭牛士 2023-10-07 11:10:50


(相关资料图)

10月7日消息,华为官方发布消息称,PyTorch社区正式发布2.1版本,经过PyTorch社区与昇腾的持续合作和共同努力,Pytorch 2.1版本已同步支持昇腾NPU,意味着开发者可直接在PyTorch 2.1上基于昇腾进行模型开发。

PyTorch在新版本中发布了更加完善的第三方设备接入机制,该特性由昇腾主导,与PyTorch社区的核心维护者共同完成。基于该特性,三方AI算力设备无需对原有框架代码进行修改就能对接PyTorch框架,昇腾也提供了官方认证的Torch NPU参考实现,可以指导三方设备便捷接入。

据悉,基于PyTorch,昇腾已经适配了BLOOM、GPT-3、LLaMA等业界主流大模型。

关键词:

相关新闻