---
docker:
  image: nvidia/cuda
  runtime: nvidia
steps:
- action: sys_shell
  bash: ['nvidia-smi']