Nina/unify gpu container yaml

AI-Hypercomputer / xpk

xpk (Accelerated Processing Kit, pronounced x-p-k,) is a software tool to help Cloud developers to orchestrate training jobs on accelerators such as TPUs and GPUs on GKE.

Apache License 2.0

81 stars 23 forks source link

Nina/unify gpu container yaml #115

Closed NinaCai closed 7 months ago

NinaCai commented 7 months ago

Fixes / Features

Unify gpu workload create yaml by using get_main_container

Testing / Documentation

Testing details.

[ y ] Tests pass
[ y ] Appropriate changes to documentation are included in the PR