昇腾社区首页
中文
注册

迁移后训练

迁移完成后的训练脚本可在NPU上运行,操作步骤参考如下:

  1. 单击菜单栏Run > Edit Configurations...
  2. 进入运行配置界面,参数设置如图1所示。参数设置完成后,单击“OK”
    1. 选择迁移后的训练脚本。
    2. 配置环境变量,其中,DEVICE_ID为训练进程在指定卡上运行时对应的卡号,请根据实际情况配置。
    图1 运行配置
  3. 单击,启动训练。

    图2 训练成功截图