gpu-openshift-p4d24xlarge.yml

apiVersion: v1
kind: Pod
metadata:
  name: pytorch-notebook
  namespace: rh-jarvis-ai--notebook
  labels:
    appcode: JARVIS-001
    paas.redhat.com/instance-size: p4d.24xlarge
    serviceowner: rh-jarvis-ai
    tenant.paas.redhat.com/tenant: rh-jarvis-ai
spec:
  restartPolicy: Never
  tolerations:
    - effect: NoSchedule
      key: tenant
      value: 'rh-jarvis-ai'
  containers:
    - name: tuhsharm-pytorch
      ports:
      - containerPort: 8888
      image: "quay.io/opendatahub-contrib/workbench-images:cuda-jupyter-pytorch-c9s-py311_2023c_latest"
      resources:
        requests:
          nvidia.com/gpu: "1"
        limits:
          nvidia.com/gpu: "1"
---
apiVersion: v1
kind: Service
metadata:
  name: pytorch-notebook-service
  namespace: rh-jarvis-ai--notebook
  labels:
    appcode: JARVIS-001
    paas.redhat.com/instance-size: p4d.24xlarge
    serviceowner: rh-jarvis-ai
    tenant.paas.redhat.com/tenant: rh-jarvis-ai
spec:
  selector:
    user: "tuhsharm"
  ports:
  - protocol: TCP
    port: 80
    targetPort: 8888

---
apiVersion: route.openshift.io/v1
kind: Route
metadata:
  namespace: rh-jarvis-ai--notebook
  name: pytorch-notebook
  labels:
    appcode: JARVIS-001
    paas.redhat.com/instance-size: p4d.24xlarge
    serviceowner: rh-jarvis-ai
    tenant.paas.redhat.com/tenant: rh-jarvis-ai
spec:
  to:
    kind: Service
    name: pytorch-notebook-service
  tls: null
  port:
    targetPort: 8888