Skip to content

Instantly share code, notes, and snippets.

@ruchim

ruchim/gpu.wdl

Created February 13, 2019 04:36
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 ruchim/687ed30776c1315222acc40c6c4caafb to your computer and use it in GitHub Desktop.
Save ruchim/687ed30776c1315222acc40c6c4caafb to your computer and use it in GitHub Desktop.
Download ZIP
Raw
gpu.wdl
workflow test_gpu_wf {
call test_gpu {}
}
task test_gpu {
command { nvidia-smi }
runtime {
gpuType: "nvidia-tesla-k80"
gpuCount: 1
zones: ["us-central1-c"]
docker: "nvidia/cuda:9.0-base"
}
output { String out = read_string(stdout()) }
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment