search

SpikeInterface / spikeinterface

A Python-based module for creating flexible and robust spike sorting pipelines.
https://spikeinterface.readthedocs.io
MIT License
500 stars 187 forks source link

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

Open caniko opened 2 years ago

caniko commented 2 years ago

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 :grin:)

alejoe91 commented 2 years ago

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