昇腾社区首页
中文
注册

安装

目前TorchAir暂未提供单独的发布包,而是作为torch_npu的三方库,随着torch_npu一起发布,因此直接取torch_npu插件安装之后,即可使用TorchAir。

torch_npu的安装请参见配置与安装

PyTorch版本为2.1.0,Ascend Extension for PyTorch(torch_npu)插件版本为6.0.RC1时,才能正常使用TorchAir。