gpu-dp.yaml

apiVersion: extensions/v1beta1
kind: Deployment
metadata:
  name: gpu
  labels:
    app: gpu
spec:
  replicas: 1
  selector:
    matchLabels:
      app: gpu
  template:
    metadata:
      labels:
        app: gpu
        driver: nvidia-gpu
    spec:
      containers:
      - name: gpu-container
        image: gcr.io/tensorflow/tensorflow:latest-gpu
        imagePullPolicy: Always
        resources:
          requests:
            alpha.kubernetes.io/nvidia-gpu: 1
          limits:
            alpha.kubernetes.io/nvidia-gpu: 1