TensorFlow pod spec

tf.json

{
    "spec": {
        "containers": [
            {
                "image": "tensorflow/tensorflow:1.2.1-devel-gpu-py3",
                "imagePullPolicy": "IfNotPresent",
                "name": "tf",
                "volumeMounts": [
                    {
                        "name": "dev-nvidia1",
                        "mountPath": "/dev/nvidia1"
                    },
                    {
                        "name": "nvidia-driver-375-66",
                        "mountPath": "/usr/local/nvidia",
                        "readOnly": true
                    },
                    {
                        "name": "libcuda-so",
                        "mountPath": "/usr/lib/x86_64-linux-gnu/libcuda.so"
                    },
                    {
                        "name": "libcuda-so-1",
                        "mountPath": "/usr/lib/x86_64-linux-gnu/libcuda.so.1"
                    },
                    {
                        "name": "libcuda-so-375-66",
                        "mountPath": "/usr/lib/x86_64-linux-gnu/libcuda.so.375.66"
                    }
                ],
                "ports": [
                    {
                        "hostPort": 10000,
                        "containerPort": 5555
                    }
                ],
                "env": [
                    {
                        "name": "hostip",
                        "value": "172.16.34.150"
                    }
                ],
                "resources": {
                    "limits": {
                        "cpu": "1.2",
                        "memory": "0.5Gi", 
                        "alpha.kubernetes.io/nvidia-gpu": "1"
                    }
                }
            }
        ],
        "restartPolicy": "Never",
        "volumes": [
            {
                "name": "dev-nvidia1",
                "hostPath": {
                    "path": "/dev/nvidia1"
                }
            },
            {
                "name": "nvidia-driver-375-66",
                "hostPath": {
                    "path": "/var/lib/nvidia-docker/volumes/nvidia_driver/375.66"
                }
            },
            {
                "name": "libcuda-so",
                "hostPath": {
                    "path": "/usr/lib/x86_64-linux-gnu/libcuda.so"
                }
            },
            {
                "name": "libcuda-so-1",
                "hostPath": {
                    "path": "/usr/lib/x86_64-linux-gnu/libcuda.so.1"
                }
            },
            {
                "name": "libcuda-so-375-66",
                "hostPath": {
                    "path": "/usr/lib/x86_64-linux-gnu/libcuda.so.375.66"
                }
            }
        ]
    },
    "apiVersion": "v1",
    "kind": "Pod",
    "metadata": {
        "name": "tf-1",
      "labels": {
         "name": "tf"
      }
    }
}