Use WSL2 on windows to sort big files

[robbeWU]

Hi,

I have a question about using Spike Interface on Windows 11 using docker images for kilosort 4.
I installed docker desktop version 27.0.3 and Ubuntu 22.04 to run docker with wsl2 backend.

When I run sorting_KS4 = ss.run_sorter('kilosort4', recording_with_probe,
output_folder=save_folder/ 'output',
verbose=True,
docker_image='spikeinterface/kilosort4-base:latest',
)
It crashes when I use very large files, but not with smaller files, reason for me to think that the problem might be GPU power.
When I run this command in PowerShell: "docker run --rm -it --gpus=all nvcr.io/nvidia/k8s/cuda-sample:nbody nbody -gpu -benchmark" (from https://docs.docker.com/desktop/gpu/) the output is as expected and so it seems docker has access to full gpu power, what do I have to do to make use of this in the run_sorter function in my python script?

Or is there an unrelated issue that I am overseeing?
I have tried to run my script in Python for Ubuntu completely, is that the only way to make use of full GPU power? The docker documentation makes me think it should also be possible on Windows, but maybe the ss.run_sorter function needs some extra input commands or?

Please help,

Thanks

[JoeZiminski]

Hey @robbeWU, I'm not so familiar with WSL so it's just a guess, but the WSL gets allocated a portion of RAM and hard disk and maybe it is getting used up with the larger file? Can you access the error message? It might be worth checking the RAM and hard drive allocation for your WSL. BTW, I might be misunderstanding your setup but you can get Docker desktop for Windows and run directly in Windows (or, even without docker for KS4 as it is written in python so you can install directly from pypi). However maybe this is for GPU driver-related stuff?

[zm711]

I think we also need to know what you mean by crash? Is there a stack trace, does it hang etc.

edit sorry just commented on your comment Joe. my bad.