hubmapconsortium / codex-pipeline

CODEX data processing code
GNU General Public License v3.0
10 stars 4 forks source link

Support "all" GPUs #24

Open mruffalo opened 3 years ago

mruffalo commented 3 years ago

GPU selection currently requires a comma-separated list of integers. Expand this to allow the string "all", after determining whether this can be passed directly to Cytokit -- if not, we'll need to do something like the GPU selection in the segmentations repository: https://github.com/hubmapconsortium/segmentations/blob/b0c35cdb482b716bd9b934744db744200e37bd4c/bin/main.py#L20