Feature request: gputouse keyword argument in run_<sorter that supports GPU>() and run_sorter()

[caniko]

Feature

gputouse: Optional[int, Sequence[int]]. Integer or Sequence of integers indicating the index of the GPU(s) for spike sorting. For CPU, and not GPU, set to None. Index for each GPU can be found with nvidia-smi command.

This keyword argument should be available on sorters that support GPU; run_<sorter that supports GPU>() and run_sorter().

Remarks
Containerized workflows can also take advantage of this kwarg. Docker supports GPU-passthrough on all major platforms; including Windows (WSL 2 required). Should be possible with Singularity.

Question
Do any of the sorters support multi-gpu?

(last issue of the day 😁)

[alejoe91]

That would be a good addition! Can you make a PR Can?