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