apiVersion: v1
kind: Pod
spec:
  securityContext:
    runAsUser: 0
  containers:
    - name: dgl-ci-gpu
      image: dgllib/dgl-ci-gpu:conda
      imagePullPolicy: Always
      tty: true
      resources:
        limits:
          nvidia.com/gpu: 1 # requesting 1 GPU
  affinity:
    nodeAffinity:
      requiredDuringSchedulingIgnoredDuringExecution:
        nodeSelectorTerms:
        - matchExpressions:
          - key: beta.kubernetes.io/instance-type
            operator: In
            values:
            - g4dn.2xlarge