fix ks4 GPU with container

SpikeInterface · Mar 15, 2024 · 6cbe4c5 · 6cbe4c5
1 parent db49738
commit 6cbe4c5
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/spikeinterface/sorters/external/kilosort4.py b/src/spikeinterface/sorters/external/kilosort4.py
@@ -15,6 +15,7 @@ class Kilosort4Sorter(BaseSorter):
 
     sorter_name: str = "kilosort4"
     requires_locations = True
+    gpu_capability = "nvidia-optional"
 
     _default_params = {
         "nblocks": 1,