Documentation
搜索文档…
Kubeflow

Kubeflow集成

使用某些功能需要额外的依赖关系。通过运行pip install wandb[kubeflow]安装所有Kubeflow的依赖关系。

训练作业

目前W&B会自动读取TF_CONFIG 环境变量,对分布式运行进行分组。

Arena

wandb库通过自动将认证信息添加到容器环境与 arena 集成。如果你想在本地使用wandb包装器,请在你的 .bashrc 中添加以下内容。
1
alias arena="python -m wandb.kubeflow.arena"
Copied!
I如果你没有在本地安装arena ,上面的命令将使用wandb/arena docker镜像,并尝试挂载你的kubectl 配置。

Pipelines

wandb提供了一个可以在pipelines中使用的arena_launcher_op
如果你想要构建你自己的自定义启动器操作,你也可以使用这段代码来添加pipeline_metadata。为了wandb认证,你应该将WANDB_API_KEY 添加到该操作中,然后你的启动器可以将相同的环境变量添加到训练容器中。
1
import osfrom kubernetes import client as k8s_client​op = dsl.ContainerOp( ... )op.add_env_variable(k8s_client.V1EnvVar( name='WANDB_API_KEY', value=os.environ["WANDB_API_KEY"]))
Copied!
最近更新 9mo ago