下载
中文
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

安装部署

如何安装PyTorch与torch_npu?

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

如何使用混合精度?

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

搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词