tf-run-drivers.yaml

apiVersion: extensions/v1beta1
kind: Deployment
metadata:
  labels:
    app: tensorflow
  name: tensorflow
spec:
  template:
    metadata:
      labels:
        app: tensorflow
    spec:     
      containers:
      - name: tensorflow
        image: tensorflow/tensorflow:latest-gpu
        command: ["python main.py"]      
        imagePullPolicy: IfNotPresent
        env:
        - name: LD_LIBRARY_PATH
          value: /usr/lib/x86_64-linux-gnu_with_nvidia
        resources:
          requests:
            alpha.kubernetes.io/nvidia-gpu: 1 
          limits:
            alpha.kubernetes.io/nvidia-gpu: 1 
        volumeMounts:
        - mountPath: /usr/lib/x86_64-linux-gnu_with_nvidia
          name: libraries
      volumes:
      - name: libraries
        hostPath:
          path: /usr/lib/x86_64-linux-gnu