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

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

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

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