apiVersion: apps/v1
kind: Deployment
metadata:
  name: gpu-presuretest-deployment
  namespace: data-infra
spec:
  replicas: 1
  selector:
    matchLabels:
      app: gpu-presuretest-deployment
  template:
    metadata:
      labels:
        app: gpu-presuretest-deployment
    spec:
      nodeSelector:
        infra.xaminim.com/machine-type: "gpu"  # 这里是上一步打的标签
      nodeName: a.a.a.a  # 替换为你的GPU节点
      containers:
      - name: gpu-container
        image: harbor.xaminim.com/minimax-infra/benchmarktool/gpu:v0.4-cuda12.2
        command: ["sleep"]
        args: ["infinity"]
        resources:
          limits:
            nvidia.com/gpu: "1"  # 请求8个GPU
      imagePullSecrets:
      - name: minimax-harbor-registry
      tolerations:
      - operator: "Exists"

