代码自动迁移

自动迁移操作简单,且修改内容少,只需在训练脚本中添加 from torch_npu.contrib import transfer_to_npu 迁移代码。

PyTorch 2.4.0及之前版本,添加如下迁移代码:

import torch
import torch_npu
...
from torch_npu.contrib import transfer_to_npu

PyTorch 2.5.1及之后版本,添加如下迁移代码:

import torch
...
from torch_npu.contrib import transfer_to_npu