Skip to content

Kubernetes GPU 任务生成 ​

在实验室的 Kubernetes 集群上跑实验,不用每次从头写 YAML。

说明 ​

  • GPU 通过 nvidia.com/gpu 申请,需要集群安装了 NVIDIA 设备插件。
  • PyTorch 的 DataLoader 多进程需要较大的共享内存,生成的配置把 /dev/shm 挂载为内存盘。
  • 提交:kubectl apply -f job.yaml;查看日志:kubectl logs -f job/任务名。