Skip to content

Instantly share code, notes, and snippets.

@wbuchwalter
Last active February 20, 2018 23:41
Show Gist options
  • Save wbuchwalter/c69ebba322781e8882f424e52833418c to your computer and use it in GitHub Desktop.
Save wbuchwalter/c69ebba322781e8882f424e52833418c to your computer and use it in GitHub Desktop.
apiVersion: batch/v1
kind: Job
metadata:
name: nvidia-smi
spec:
template:
metadata:
name: nvidia-smi
spec:
restartPolicy: Never
containers:
- name: nvidia-smi
image: nvidia/cuda
command:
- nvidia-smi
resources:
limits:
alpha.kubernetes.io/nvidia-gpu: 1
volumeMounts:
- name: nvidia
mountPath: /usr/local/nvidia/
volumes:
- name: nvidia
hostPath:
path: /usr/local/nvidia
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment