Skip to content

Instantly share code, notes, and snippets.

@sadysnaat

sadysnaat/cuda-add-pod.yaml

Created August 10, 2020 07:23
Show Gist options
  • Save sadysnaat/5f6badceae19aa26850a30957a7327bf to your computer and use it in GitHub Desktop.
Save sadysnaat/5f6badceae19aa26850a30957a7327bf to your computer and use it in GitHub Desktop.
Download ZIP
Raw
cuda-add-pod.yaml
apiVersion: v1
kind: Pod
metadata:
name: cuda-vector-add
spec:
restartPolicy: OnFailure
containers:
- name: cuda-vector-add
image: "gcr.io/spectro-images-public/gpu/nvidia/ubuntu-nvidia-add:ubuntu
# specify gpu request
resources:
limits:
nvidia.com/gpu: 1 # requesting 1 GPU
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment