下载
EN
注册

亮点特性

继承开源PyTorch特性,适配昇腾AI处理器系列;实现使用昇腾AI处理器进行动态图训练,现有网络只修改设备类型和数据类型,即可迁移到昇腾AI处理器上使用;支持自定义算子开发和模型调整
适配昇腾AI处理器
基于开源PyTorch,适配昇腾AI处理器,提供原生Python接口
框架基础功能
PyTorch动态图、自动微分、Profiling、优化器等
自定义算子开发
支持在PyTorch框架中添加自定义算子
分布式训练
支持原生分布式数据并行训练,包含单机多卡、多机多卡场景支持的集合通信原语包含Broadcast、AllReduce等
混合精度
支持O1、O2混合精度和静态动态Loss Scale算法,并且能够通过黑白名单配置算子的精度。立即下载、安装Apex混合精度优化包
模型推理
支持输出标准的ONNX模型可通过离线转换工具将ONNX模型转换为离线推理模型

学习成长资源

快速上手 轻松入门
快速上手PyTorch框架,完成环境搭建并进行迁移训练体验
迁移训练 全面掌握
全面掌握PyTorch框架,完成模型的迁移及训练
训练调优 深度分析
全场景、多角度精度与性能分析,助力用户成长