我要评分文档获取效率文档正确性内容完整性文档易理解0/200提交在线提单论坛求助昇腾小AI 安装部署 如何安装PyTorch与torch_npu? 目前以安装CANN开发套件包前提下,原生PyTorch框架与昇腾开发的Ascend Extension for PyTorch插件配合使用。用户需下载对应的whl安装包或使用源码编译的方式,安装框架和框架插件,在代码中添加import torch与import torch_npu以引入框架和框架插件。详细安装与使用请参见《Ascend Extension for PyTorch 配置与安装》完成开发套件包的安装与框架和插件的安装。 如何使用混合精度? 用户可使用PyTorch框架插件自带的AMP功能模块或第三方APEX混合精度模块。推荐用户使用框架自带的AMP混合精度模块。 详细使用请参见《PyTorch 训练模型迁移调优指南》中的“迁移适配 > 模型训练适配 > 关键特性适配 > 混合精度适配(可选)”章节。 搜索结果找到“0”个结果当前产品无相关内容未找到相关内容,请尝试其他搜索词