Kilosort 4 & CUDA & NVIDIA compatilbilty

Fails at first clustering

[2025-10-22 15:44:00,163] ERROR: ERROR in CDKL5-E6D_T1_C1,M07039 240604 70 well000/rec0000: Spike sorting error trace: Traceback (most recent call last): File "/home/mmp/.local/lib/python3.10/site-packages/spikeinterface/sorters/basesorter.py", line 270, in run_from_folder SorterClass._run_from_folder(sorter_output_folder, sorter_params, verbose) File "/home/mmp/.local/lib/python3.10/site-packages/spikeinterface/sorters/external/kilosort4.py", line 380, in _run_from_folder st, tF, _, _ = detect_spikes(**detect_spikes_kwargs) File "/home/mmp/.local/lib/python3.10/site-packages/kilosort/run_kilosort.py", line 797, in detect_spikes Wall3 = template_matching.postprocess_templates( File "/home/mmp/.local/lib/python3.10/site-packages/kilosort/template_matching.py", line 144, in postprocess_templates Wall2, _ = align_U(Wall, ops, device=device) File "/home/mmp/.local/lib/python3.10/site-packages/kilosort/template_matching.py", line 132, in align_U Xmax = X.abs().max(0)[0].max(0)[0].reshape(-1, ops['nt']) torch.OutOfMemoryError: CUDA out of memory. Tried to allocate 6.62 GiB. GPU 0 has a total capacity of 11.76 GiB of which 3.77 GiB is free. Including non-PyTorch memory, this process has 7.98 GiB memory in use. Of the allocated memory 7.75 GiB is allocated by PyTorch, and 56.60 MiB is reserved by PyTorch but unallocated. If reserved but unallocated memory is large try setting PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True to avoid fragmentation. See documentation for Memory Management (https://pytorch.org/docs/stable/notes/cuda.html#environment-variables)

Spike sorting failed. You can inspect the runtime trace in /mnt/disk15tb/mmpatil/MEA_Analysis_SIUpgrade/IPNAnalysis/../AnalyzedData/CDKL5-E6D_T1_C1/CDKL5-E6D_T1_C1/240604/M07039/Network/000070/well000/analyzer_output/spikeinterface_log.json. - mea_analysis_routine - 700 [2025-10-22 15:44:00,163] ERROR: TRACEBACK: Traceback (most recent call last): File "/mnt/disk15tb/mmpatil/MEA_Analysis_SIUpgrade/IPNAnalysis/mea_analysis_routine.py", line 598, in process_block sorting_obj = si.run_sorter_local( File "/home/mmp/.local/lib/python3.10/site-packages/spikeinterface/sorters/runsorter.py", line 302, in run_sorter_local SorterClass.run_from_folder(folder, raise_error, verbose) File "/home/mmp/.local/lib/python3.10/site-packages/spikeinterface/sorters/basesorter.py", line 310, in run_from_folder raise SpikeSortingError( spikeinterface.sorters.utils.misc.SpikeSortingError: Spike sorting error trace: Traceback (most recent call last): File "/home/mmp/.local/lib/python3.10/site-packages/spikeinterface/sorters/basesorter.py", line 270, in run_from_folder SorterClass._run_from_folder(sorter_output_folder, sorter_params, verbose) File "/home/mmp/.local/lib/python3.10/site-packages/spikeinterface/sorters/external/kilosort4.py", line 380, in _run_from_folder st, tF, _, _ = detect_spikes(**detect_spikes_kwargs) File "/home/mmp/.local/lib/python3.10/site-packages/kilosort/run_kilosort.py", line 797, in detect_spikes Wall3 = template_matching.postprocess_templates( File "/home/mmp/.local/lib/python3.10/site-packages/kilosort/template_matching.py", line 144, in postprocess_templates Wall2, _ = align_U(Wall, ops, device=device) File "/home/mmp/.local/lib/python3.10/site-packages/kilosort/template_matching.py", line 132, in align_U Xmax = X.abs().max(0)[0].max(0)[0].reshape(-1, ops['nt']) torch.OutOfMemoryError: CUDA out of memory. Tried to allocate 6.62 GiB. GPU 0 has a total capacity of 11.76 GiB of which 3.77 GiB is free. Including non-PyTorch memory, this process has 7.98 GiB memory in use. Of the allocated memory 7.75 GiB is allocated by PyTorch, and 56.60 MiB is reserved by PyTorch but unallocated. If reserved but unallocated memory is large try setting PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True to avoid fragmentation. See documentation for Memory Management (https://pytorch.org/docs/stable/notes/cuda.html#environment-variables)

Spike sorting failed. You can inspect the runtime trace in /mnt/disk15tb/mmpatil/MEA_Analysis_SIUpgrade/IPNAnalysis/../AnalyzedData/CDKL5-E6D_T1_C1/CDKL5-E6D_T1_C1/240604/M07039/Network/000070/well000/analyzer_output/spikeinterface_log.json.

• mea_analysis_routine - 701