介绍
本文主要向用户介绍鲲鹏服务器上Transformers的开发指南,包含源码构建、安装和基于PyTorch CUDA后端的基础能力验证场景。
Transformers提供了统一的API与数千个预训练模型,覆盖从模型加载、训练微调到推理部署的全生命周期,并原生支持PyTorch、TensorFlow与JAX三大主流深度学习框架。 在鲲鹏服务器上部署Transformers,不仅可通过昇腾CANN生态获得针对多头注意力等核心算子的深度硬件加速,还能依托开源社区的ARM64预编译优化实现无缝迁移。
父主题: 开发指南