apiVersion: v1
kind: Pod
metadata:
  name: gpu-pod-caffe
spec:
  containers:
    - name: caffe
      image: jammingyu9/ai:caffe_nvidia
      workingDir: /root
      command: ["/bin/bash", "-c", "--"]
      args: ["sleep infinity & wait"]
      resources:
        limits:
          nvidia.com/gpu: 1 # requesting 1 GPUs
