• davincirunsdk
• • • 调试环境（开发环境）
    • 训练作业
    • AI靶场全量运行
  • LICENSE
  • 开发指南
  • • 克隆并安装
    • 单元测试
    • 项目构成
  • 案例
  • 鸣谢

davincirunsdk

为类Jupyter交互式环境提供Notebook友好的Ascend分布式训练SDK

调试环境（开发环境）

以MindSpore1.5分布式训练教程
为例，使用本SDK可改造为

import os

os.environ['DATA_PATH'] = '/cache/cifar-10-batches-bin'
from davincirunsdk import start_distributed_train, wait_distributed_train

cmd = ['python', 'resnet50_distributed_training.py']
manager = start_distributed_train(cmd, output_notebook=True)
wait_distributed_train(manager)

训练作业

davincirun train.py

或在python文件中使用：

from davincirunsdk import init_rank_table, start_distributed_train, wait_distributed_train

init_rank_table()
manager = start_distributed_train(['python', 'train.py'])
wait_distributed_train(manager)

AI靶场全量运行

同调试环境（开发环境），不需要额外修改

更多细节见案例

LICENSE

MIT License

开发指南

克隆并安装

$git clone https://git.openi.org.cn/Wh1isper/davincirunsdk.git
$cd davincirunsdk
$pip install -e ./

单元测试

$pytest .

项目构成

notebook文件夹下是针对notebook运行环境修改的davincirun文件，以及sdk入口

davincirunsdk目录下，除了notebook外的文件，是原有davincurun代码，进行了python包改造

案例

AI靶场分布式训练支持

鸣谢

感谢华为云、鹏城实验室、AI靶场对本项目的大力支持和帮助