Skip to content
科研工具箱
搜索
K
Main Navigation
科研工具
实用工具
关于
相关站点
实验室主页
科研入门手册
科研技能包
深色模式
目录
回到顶部
本页目录
Kubernetes GPU 任务生成
在实验室的 Kubernetes 集群上跑实验,不用每次从头写 YAML。
说明
GPU 通过 nvidia.com/gpu 申请,需要集群安装了 NVIDIA 设备插件。
PyTorch 的 DataLoader 多进程需要较大的共享内存,生成的配置把 /dev/shm 挂载为内存盘。
提交:kubectl apply -f job.yaml;查看日志:kubectl logs -f job/任务名。