Skip to content

Instantly share code, notes, and snippets.

@stantonk

stantonk/gpu-test-vector-add.yaml

Created April 10, 2023 20:45
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save stantonk/679637e75ed3cea3f17c5ab54902bddb to your computer and use it in GitHub Desktop.
Save stantonk/679637e75ed3cea3f17c5ab54902bddb to your computer and use it in GitHub Desktop.
Download ZIP
Raw
gpu-test-vector-add.yaml
apiVersion: v1
kind: Pod
metadata:
name: gpu-test-vector-add
spec:
runtimeClassName: nvidia
restartPolicy: Never
containers:
- name: cuda-container
image: nvcr.io/nvidia/k8s/cuda-sample:vectoradd-cuda10.2
resources:
limits:
nvidia.com/gpu: 1 # requesting 1 GPU
tolerations:
- key: nvidia.com/gpu
operator: Exists
effect: NoSchedule
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment