change gpu test pod to run the cuda runtime on pascal
This commit is contained in:
parent
5a2648c448
commit
eb3068048e
24
nvidia/gpu-pod.yaml
Normal file
24
nvidia/gpu-pod.yaml
Normal 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
|
|
@ -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
|
|
Loading…
Reference in New Issue
Block a user