apiVersion: v1
kind: Pod
metadata:
  name: gpu-pod
spec:
  containers:
    - name: cuda-container
      image: hxhp/elastic-horovod-demo:1.0
      #resources:
      #  limits:
      #    alnair/vgpu-memory: 2
      #    alnair/vgpu-compute: 50 
      command: ["python", "/examples/tensorflow2_mnist.py"]
  restartPolicy: OnFailure
