首页
产品
解决方案
开发者
合作伙伴
高校
昇腾AI市场
产业资讯
支持与服务
产品
昇腾计算产业概述
资源下载中心
固件与驱动
Atlas系列硬件
开发者套件
加速模块
加速卡
智能小站
服务器
集群
异构计算架构
CANN
AI框架
昇思MindSpore
PyTorch框架适配
TensorFlow框架适配
应用使能
MindSpeed
MindIE
MindCluster
MindEdge
MindSDK
ModelZoo
全流程开发工具链
MindStudio
集群自智引擎
CCAE
解决方案
行业应用
人工智能计算中心
能源
金融
交通
电信
智慧城市
制造
医疗
解决方案
昇腾大模型解决方案
城市智能中枢
昇腾生态解决方案
开发者
开发者主页
论坛
鲲鹏昇腾原生创新汇
开发场景
推理开发
NEW
算子开发
NEW
开发资源
资源下载中心
固件与驱动
昇腾镜像仓库
兼容性查询助手
ModelZoo
昇腾开放资源
专区
开发板专区
HOT
DeepSeek专区
HOT
文档
活动
鲲鹏昇腾开发者大会2025
HOT
昇腾AI开发者创享日
昇腾学堂·活动广场
昇腾AI训练营-CANN训练营
HOT
大赛
昇腾AI创新大赛2025
昇腾AI原生创新算子挑战赛
HOT
直播
学习
学习路径
在线课程
在线实验
认证
互动
论坛
博客
NEW
技术干货
昇腾社区APP
开发者计划
昇腾开发者计划2025
昇腾众智计划
昇腾荣誉顾问
昇腾MVP
昇腾HAE
原生创新汇会员
鲲鹏昇腾原生创新汇
NEW
合作伙伴
登录生态伙伴中心
合作伙伴
昇腾生态合作伙伴网络
昇腾万里伙伴计划
昇腾技术认证书查询
昇腾兼容性认证硬件目录
高校
高校人才发展主页
智能基座
教学资源
师资培训
实践创新
智能基座社团
虚拟教研室
人才发展计划
鲲鹏&昇腾产教融合育人基地
鲲鹏&昇腾本科教学推广计划
虚拟教研室教改专项计划
计算高校开源人才奖励计划
鲲鹏&昇腾原生人才高校促进计划
昇腾众智计划
鲲鹏昇腾科研创新使能计划
鲲鹏昇腾科教创新卓越中心/孵化中心
资源
学习资源
活动与大赛
华为全联接⼤会2024⾼校系列活动
鲲鹏昇腾科教创新卓越中心/孵化中心特训营
中国国际大学生创新大赛2024
国家级大学生创新训练计划-华为命题2024
昇腾AI创新大赛2025
华为全联接大会2023高校系列活动
第六届中国软件开源创新大赛
华为软件精英挑战赛2023
高校资讯
新闻资讯
优秀展播
昇腾AI市场
市场主页
市场分类
解决方案
原生大模型
ModelZoo
应用案例
Powered by Ascend
产业资讯
活动
鲲鹏昇腾开发者大会2025
HOT
WAIC 昇腾人工智能产业高峰论坛2024
资讯
新闻资讯
支持与服务
支持与服务
文档
技术支持
论坛求助
技术工单
自助查询
常见问题
故障案例
文档
昇腾论坛
资讯
产品公告
文档
HOT
下载
中文
注册
登录
PyTorch框架适配
暂无评分
我要评分
该页面对您有帮助吗?
我要评分
昇腾PyTorch适配版本,和社区同步发布和演进,支持PyTorch主流三方生态
立即下载
查看文档
获取源码
适配昇腾AI处理器
基于开源PyTorch,适配昇腾AI处理器,提供原生Python接口
框架基础功能
PyTorch动态图、自动微分、Profiling、优化器等
自定义算子开发
支持在PyTorch框架中添加自定义算子
分布式训练
支持原生分布式数据并行训练,包含单机多卡、多机多卡场景支持的集合通信原语包含Broadcast、AllReduce等
混合精度
支持O1、O2混合精度和静态动态Loss Scale算法,并且能够通过黑白名单配置算子的精度。立即下载、安装Apex混合精度优化包
模型推理
支持输出标准的ONNX模型,可通过离线转换工具将ONNX模型转换为离线推理模型
开发场景
模型脚本适配
通过自动化迁移脚本或少量手动迁移,适配昇腾自有API接口,将其他人工智能平台的模型脚本迁移至昇腾AI平台进行模型训练,契合昇腾AI处理器
模型训练
基于原生PyTorch框架API和昇腾自有API,开发和训练网络模型,并利用昇腾自研调优工具对训练精度性能进行深度优化,获得高精度高性能人工智能算法
算子开发
通过昇腾提供的开发工具链,开发基于昇腾AI处理器的自定义算子,实现网络模型深度优化
开发资源
PyTorch模型库
支持原生分布式数据并行训练,包含单机多卡、多机多卡场景支持的集合通信原语包含Broadcast、AllReduce等
获取模型
学习资源
快速上手 轻松入门
快速上手PyTorch框架,完成环境搭建并进行迁移训练体验
迁移训练 全面掌握
全面掌握PyTorch框架,完成模型的迁移及训练
训练调优 深度分析
全场景、多角度精度与性能分析,助力用户成长
支持与服务
昇腾论坛
问题求助,技术探讨,大咖在线答疑解惑
支持与服务
为您提供多样化的自助查询途径、专业高效的人工服务支持