自定义框架

创建任务时 训练框架 选 自定义。仅在无法用 PyTorch、MPI、TensorFlow 三种角色模板时使用。通用字段见 创建任务配置说明。

角色与节点

可增删角色，至少保留 1 个。每个角色可选 GPU 或 CPU。平台不强制各角色卡数一致。

平台不为自定义框架写入 PyTorch 或 TensorFlow 那种专用变量，只写入各节点都能读到的通用变量。名单见 通用环境变量列表。节点如何互相发现，由启动命令和代码自己完成。

启动命令

查看示例 仍可看 PyTorch、MPI、TensorFlow 三份作参考。选自定义不会自动填入。

当前占位示例的运行命令为 bash。运行参数：

-c
python /workspace/train.py

改成您的启动器和容器内绝对路径。镜像里没有 bash 时，改掉运行命令。