安装部署

如何安装PyTorch与torch_npu？

目前以安装CANN开发套件包前提下，原生PyTorch框架与昇腾开发的Ascend Extension for PyTorch插件配合使用。用户需下载对应的whl安装包或使用源码编译的方式，安装框架和框架插件，在代码中添加import torch与import torch_npu以引入框架和框架插件。详细安装与使用请参见《Ascend Extension for PyTorch 配置与安装》完成开发套件包的安装与框架和插件的安装。

如何使用混合精度？

用户可使用PyTorch框架插件自带的AMP功能模块或第三方APEX混合精度模块。推荐用户使用框架自带的AMP混合精度模块。 详细使用请参见《PyTorch 训练模型迁移调优指南》中的“迁移适配 > 模型训练适配 > 关键特性适配 > 混合精度适配（可选）”章节。