github.com/sentienttechnologies/studio-go-runner@v0.0.0-20201118202441-6d21f2ced8ee/examples/azure/test-azure-gpu.yaml (about)

     1  # Copyright (c) 2019-2020 Cognizant Digital Business, Evolutionary AI. All rights reserved. Issued under the Apache 2.0 License.
     2  apiVersion: batch/v1
     3  kind: Job
     4  metadata:
     5    labels:
     6      app: samples-tf-mnist-demo
     7    name: samples-tf-mnist-demo
     8  spec:
     9    template:
    10      metadata:
    11        labels:
    12          app: samples-tf-mnist-demo
    13      spec:
    14        containers:
    15        - name: samples-tf-mnist-demo
    16          image: microsoft/samples-tf-mnist-demo:gpu
    17          args: ["--max_steps", "50000"]
    18          imagePullPolicy: IfNotPresent
    19          resources:
    20            limits:
    21             nvidia.com/gpu: 1
    22        restartPolicy: OnFailure