第三方框架适配
注册
第三方框架适配PyTorch
PyTorch
TensorFlow
PyTorch
TensorFlow

PyTorch

继承开源PyTorch特性,适配昇腾AI处理器系列;实现使用昇腾AI处理器进行动态图训练,现有网络只修改设备类型和数据类型,即可迁移到昇腾AI处理器上使用;支持自定义算子开发和模型调整

亮点特性

适配昇腾 AI处理器
框架 基础功能
自定义 算子开发
分布式 训练
混合 精度
模型 推理

学习成长资源

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

获取开发支持与帮助