diff --git a/kilosort4/Dockerfile b/kilosort4/Dockerfile index 5306ea6..e139b3a 100644 --- a/kilosort4/Dockerfile +++ b/kilosort4/Dockerfile @@ -35,4 +35,4 @@ RUN conda init bash RUN pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118 # install kilosort4 -RUN pip install kilosort==4.0.12 +RUN pip install kilosort==4.0.16 diff --git a/kilosort4/build.sh b/kilosort4/build.sh index 79095a1..8f7e3ca 100755 --- a/kilosort4/build.sh +++ b/kilosort4/build.sh @@ -1,3 +1,3 @@ #!/bin/bash -docker build -t spikeinterface/kilosort4-base:latest -t spikeinterface/kilosort4-base:4.0.12_cuda-12.0.0 . +docker build -t spikeinterface/kilosort4-base:latest -t spikeinterface/kilosort4-base:4.0.16_cuda-12.0.0 .