apiVersion: apps/v1
kind: Deployment
metadata:
  name: tf-notebook
spec:
  replicas: 2
  selector:
    matchLabels:
      app: tf-notebook
      user: user1
  template:
    metadata:
      labels:
        app: tf-notebook
        user: user1
    spec:
      nodeSelector:
        nodegroup: gpu
      containers:
        - name: tf-notebook
          image: tensorflow/tensorflow:2.16.2-gpu-jupyter
          resources:
            limits:
              cpu: "4"
              memory: 16Gi
              nvidia.com/gpu: 2
          ports:
            - containerPort: 8888
              name: notebook
          volumeMounts:
            - name: notebook-storage
              mountPath: /sample
      volumes:
        - name: notebook-storage
          persistentVolumeClaim:
            claimName: tf-notebook-pvc
---
apiVersion: v1
kind: Service
metadata:
  name: tf-notebook
  labels:
    app: tf-notebook
    user: user1
spec:
  type: ClusterIP
  ports:
    - port: 80
      name: http
      targetPort: 8888
  selector:
    app: tf-notebook
    user: user1