diff --git a/kilosort4/Dockerfile b/kilosort4/Dockerfile index 5f64372..5306ea6 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.6 +RUN pip install kilosort==4.0.12 diff --git a/kilosort4/build.sh b/kilosort4/build.sh index ebb5cae..79095a1 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.6_cuda-12.0.0 . +docker build -t spikeinterface/kilosort4-base:latest -t spikeinterface/kilosort4-base:4.0.12_cuda-12.0.0 .