安装PyTorch
若用户仅进行离线推理,请跳过此章节。
为使用PyTorch框架的开发者提供昇腾AI处理器的超强算力,昇腾开发Ascend Extension for PyTorch插件用于适配PyTorch框架,请参考《Ascend Extension for PyTorch 配置与安装》的如下章节在昇腾环境安装PyTorch框架和Ascend Extension for PyTorch插件。
- 安装前必读
- 安装PyTorch框架
- 安装torch_npu插件。
- 安装APEX模块。
父主题: 附录B:常用操作