apiVersion: v1
kind: Pod
metadata:
  name: gpu-pod-00
  namespace: pai-gpu
  annotations:
    alibabacloud.com/gpu-mem: 1024Mi
    alibabacloud.com/gpu-count: "1"
spec:
    containers:
    - image: tensorflow:latest
      name: main
      resources:
        limits:
          cpu: 4
          memory: 9216Mi
        requests:
          cpu: 4
          memory: 9216Mi