代码自动迁移
自动迁移操作简单,且修改内容少,只需在训练脚本中添加 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