change gpu test pod to run the cuda runtime on pascal

This commit is contained in:
James Andariese 2023-12-24 14:02:37 -06:00
parent 5a2648c448
commit eb3068048e
2 changed files with 24 additions and 22 deletions

24
nvidia/gpu-pod.yaml Normal file
View File

@ -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

View File

@ -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