38 lines
627 B
YAML
38 lines
627 B
YAML
nameOverride: "nccl-test"
|
|
replicaCount: 2
|
|
|
|
image:
|
|
repository: nvcr.io/nvidia/pytorch
|
|
tag: "24.10-py3"
|
|
pullPolicy: IfNotPresent
|
|
|
|
training:
|
|
nprocPerNode: 8
|
|
masterPort: 29500
|
|
batchSize: 1200
|
|
epochs: 90
|
|
lr: 0.01
|
|
|
|
resources:
|
|
gpu: 8
|
|
hostdev: 2
|
|
|
|
shmSize: "128Gi"
|
|
|
|
network:
|
|
nadName: "hostdevice-net"
|
|
|
|
nccl:
|
|
debug: "INFO"
|
|
debugSubsys: "INIT,NET,IB"
|
|
ibDisable: 0
|
|
socketIfname: "net"
|
|
ibHCA: "" # 비어있으면 설정 안 함
|
|
debugFile: "" # 비어있으면 설정 안 함
|
|
|
|
volumes:
|
|
infiniband:
|
|
hostPath: "/dev/infiniband"
|
|
dataset:
|
|
hostPath: "/home/ubuntu/cifar-10-batches-py"
|