apiVersion: v1
kind: Pod
metadata:
name: cuda-test
namespace: default
spec:
containers:
- name: cuda-container
image: nvidia/cuda:10.0-base
command: ["nvidia-smi"]
resources:
limits:
nvidia.com/gpu: 1