支持用户基于PyTorch框架和torch_npu插件在昇腾NPU上使用图模式进行训练和推理

源码资源网

支持用户基于PyTorch框架和torch_npu插件在昇腾NPU上使用图模式进行训练和推理
继承自PyTorch框架Dynamo模式，将PyTorch的FX图转换为GE计算图，并提供了GE计算图在昇腾NPU的编译与执行的能力。

快速上手

如果您在配置时未指定Ascend sdk的安装路径，则无法执行CPU上的调试，需要在NPU环境上进行测试。

CPU调试时，需要设置LD_LIBRARY_PATH到生成的fake so文件目录以及sdk目录

tools/env.sh会根据配置生成对应的LD_LIBRARY_PATH（如果您在配置时指定了Ascend sdk安装路径）

1. source tools/env.sh

复制代码

执行以下python脚本快速验证基本功能

1. import torch
   
2. import torchair
   
3. 
   
4. config = torchair.CompilerConfig()
   
5. npu_backend = torchair.get_npu_backend(compiler_config=config)
   
6. 
   
7. class Model(torch.nn.Module):
   
8.     def __init__(self):
   
9.         super().__init__()
   
10. 
    
11.     def forward(self, x, y):
    
12.         return torch.add(x, y)
    
13. 
    
14. model = Model()
    
15. model = torch.compile(model, backend=npu_backend, dynamic=False)
    
16. x = torch.randn(2, 2)
    
17. y = torch.randn(2, 2)
    
18. model(x, y)

复制代码

版本配套表[td]

TorchAir版本	PyTorch版本	torch_npu版本	CANN版本	Python版本
master（主线）	2.1.0	在研版本	在研版本	Python3.8.x Python3.9.x Python3.10.x
6.0.rc3	2.1.0	6.0.rc3	8.0.rc3	Python3.8.x Python3.9.x Python3.10.x
6.0.rc2	2.1.0	6.0.rc2	8.0.rc2	Python3.8.x Python3.9.x Python3.10.x
6.0.rc1	2.1.0	6.0.rc1	8.0.rc1	Python3.8.x Python3.9.x Python3.10.x

支持的型号

• Atlas A2 训练系列产品
• Atlas 推理系列产品（配置Ascend 310P AI处理器）
  

链接: https://pan.baidu.com/s/18igfjB4yoGmUr2mx6I_xoA

提取码下载：

文件名称:提取码下载.txt 
下载次数:0  文件大小:16 Bytes  售价:29金钱 [记录]
下载权限: 不限 [购买VIP]   [充值]   [在线充值]   【VIP会员6折；永久VIP4折】

安全检测，请放心下载