Problem pulling sorter docker image

[alejoss30]

Im currently trying to pull an image from the spikeinterface Docker Hub using the following code

sorting_ks3 = ss.run_sorter(sorter_name= 'kilosort3',
recording=recording_saved,
output_folder=output_folder,
verbose=True, docker_image= 'spikeinterface/kilosort3-compiled-base:latest",
)

However, after working for a few minutes, it gives back the following error

ImageNotFound: 404 Client Error for http+docker://localhost/v1.43/images/spikeinterface/kilosort3-compiled-base:latest/json: Not Found ("No such image: spikeinterface/kilosort3-compiled-base:latest")

Would appreciate any ideas on how to solve it. Thanks.

[alejoe91]

Hi, can you try to just use docker_image=True? The latest image is used by default.

Btw, does it work to pull the image from the docker API?

From terminal:
docker pull spikeinterface/kilosort3-compiled-base:latest

[alejoss30]

Hi. using

docker pull spikeinterface/kilosort3-compiled-base:latest from terminal i get pulls complete and finally get:
Status: Downloaded newer image for spikeinterface/kilosort3-compiled-base:latest

However when then i run

sorting_ks3 = ss.run_sorter(sorter_name= 'kilosort3',
recording=recording_saved,
output_folder=output_folder,
verbose=True, docker_image= True,
)

i get

APIError: 500 Server Error for http+docker://localhost/v1.43/containers/161847723c3696773db92a603af323825560a98c0a307447e2a30c28faa37c8b/start: Internal Server Error ("could not select device driver "" with capabilities: [[gpu]]")

[alejoe91]

Do you have a GPU? What is are you using?

[alejoss30]

I do have a GPU, it's a NVIDIA rtx 2060 which to my understanding also has CUDA capabilities. Running everything on Linux

[alejoe91]

ok, did you follow these instructions? https://spikeinterface.readthedocs.io/en/latest/modules/sorters.html#running-sorters-in-docker-singularity-containers

You need some additional steps to let Docker use the GPU