Skip to content

Instantly share code, notes, and snippets.

@czotti
Last active June 29, 2018 17:54
Show Gist options
  • Save czotti/dab11bee179242ec19563ddafb42fcbd to your computer and use it in GitHub Desktop.
Save czotti/dab11bee179242ec19563ddafb42fcbd to your computer and use it in GitHub Desktop.
Nextflow pipeline to test if the CUDA_VISIBLE_DEVICES is set.
#!/usr/bin/env nextflow
modes = [0, 1, 2, 3, 4]
process gpu_output {
echo true
script:
"""
echo \$CUDA_VISIBLE_DEVICES
"""
}
profiles {
gpu {
singularity.runOptions = '--nv'
process.queue = 'gpu'
process.clusterOptions = '--gres=gpu:gtx:1'
env.CUDA_VISIBLE_DEVICES='\$CUDA_VISIBLE_DEVICES'
}
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment