代码自动迁移
==================

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

PyTorch 2.4.0及之前版本，添加如下迁移代码：

.. code-block:: python

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

PyTorch 2.5.1及之后版本，添加如下迁移代码：

.. code-block:: python

   import torch 
   ... 
   from torch_npu.contrib import transfer_to_npu