From eb3068048ee6565b91431fdedec27dd3a3f95009 Mon Sep 17 00:00:00 2001 From: James Andariese Date: Sun, 24 Dec 2023 14:02:37 -0600 Subject: [PATCH] change gpu test pod to run the cuda runtime on pascal --- nvidia/gpu-pod.yaml | 24 ++++++++++++++++++++++++ nvidia/gpu-test.yaml | 22 ---------------------- 2 files changed, 24 insertions(+), 22 deletions(-) create mode 100644 nvidia/gpu-pod.yaml delete mode 100644 nvidia/gpu-test.yaml diff --git a/nvidia/gpu-pod.yaml b/nvidia/gpu-pod.yaml new file mode 100644 index 0000000..2cbe5f7 --- /dev/null +++ b/nvidia/gpu-pod.yaml @@ -0,0 +1,24 @@ +--- +apiVersion: apps/v1 +kind: DaemonSet +metadata: + namespace: default + name: gpu-pod +spec: + selector: + matchLabels: + app: gpu-pod + template: + metadata: + labels: + app: gpu-pod + spec: + nodeSelector: + nvidia.com/gpu.family: pascal + containers: + - name: gpu-pod + image: nvidia/cuda:12.3.1-runtime-ubuntu20.04 + command: ["sleep", "infinity"] + terminationGracePeriodSeconds: 0 + runtimeClassName: nvidia + restartPolicy: Always diff --git a/nvidia/gpu-test.yaml b/nvidia/gpu-test.yaml deleted file mode 100644 index 57d8a57..0000000 --- a/nvidia/gpu-test.yaml +++ /dev/null @@ -1,22 +0,0 @@ ---- -apiVersion: apps/v1 -kind: Deployment -metadata: - namespace: default - name: gpu-test -spec: - replicas: 0 - selector: - matchLabels: - app: gpu-test - strategy: - type: Recreate - template: - metadata: - labels: - app: gpu-test - spec: - containers: - - name: gpu-test - image: xychelsea/ffmpeg-nvidia:latest-jupyter - restartPolicy: Always