安装说明
为使用PyTorch框架的开发者提供昇腾AI处理器的超强算力,昇腾开发Ascend Extension for PyTorch(即torch_npu插件)用于适配PyTorch框架。
本文主要向用户介绍如何快速完成PyTorch框架、Ascend Extension for PyTorch(即torch_npu插件)以及扩展模块的安装。
硬件配套和支持的操作系统
产品 |
是否支持(训练场景) |
---|---|
√ |
|
x |
|
√ |
|
x |
|
x |
|
x |
|
√ |
安装方式
本手册提供了二进制软件包、源码编译以及二进制软件包(abi1版本)安装方式,用户可根据实际需求自行选择安装PyTorch框架和torch_npu插件的方式,不要求两者安装方式统一。