Update Kilosort-compiled images

Author: alejoe91

kilosort-compiled/compile_and_build_all_ks.sh

@@ -23,6 +23,7 @@ KILOSORT2_PATH=${3%/}
 KILOSORT25_PATH=${4%/}
 KILOSORT3_PATH=${5%/}
 
+# export MW_NVCC_PATH=/usr/local/cuda-11.2/bin/
 export MW_NVCC_PATH=/usr/bin/
 export MW_ALLOW_ANY_CUDA=1
 

kilosort-compiled/kilosort-compiled/Dockerfile

@@ -13,10 +13,11 @@ RUN apt-get update && \
     apt-get clean && \
     rm -rf /var/lib/apt/lists/*
 
-ENV MINICONDA_VERSION 23.1.0-1
-ENV PYTHON_VERSION py39
-ENV CONDA_DIR /home/miniconda3
-ENV LATEST_CONDA_SCRIPT "Miniconda3-${PYTHON_VERSION}_${MINICONDA_VERSION}-Linux-x86_64.sh"
+# install miniconda
+ENV MINICONDA_VERSION=25.5.1-0
+ENV PYTHON_VERSION=py312
+ENV CONDA_DIR=/home/miniconda3
+ENV LATEST_CONDA_SCRIPT="Miniconda3-${PYTHON_VERSION}_${MINICONDA_VERSION}-Linux-x86_64.sh"
 
 RUN wget --quiet https://repo.anaconda.com/miniconda/$LATEST_CONDA_SCRIPT -O ~/miniconda.sh && \
     bash ~/miniconda.sh -b -p $CONDA_DIR && \

kilosort-compiled/kilosort-compiled/build.sh

@@ -1,4 +1,4 @@
 #!/bin/bash
 
-docker build -t spikeinterface/kilosort-compiled-base:latest -t spikeinterface/kilosort-compiled-base:0.2.0 .
+docker build -t spikeinterface/kilosort-compiled-base:latest -t spikeinterface/kilosort-compiled-base:1.0.0_cuda-11.2.2 .
 

kilosort-compiled/kilosort2-compiled/build.sh

@@ -1,4 +1,4 @@
 #!/bin/bash
 
-docker build -t spikeinterface/kilosort2-compiled-base:latest -t spikeinterface/kilosort2-compiled-base:0.2.0 .
+docker build -t spikeinterface/kilosort2-compiled-base:latest -t spikeinterface/kilosort2-compiled-base:2.0.2_cuda-11.2.2 .
 

kilosort-compiled/kilosort2_5-compiled/Dockerfile

@@ -13,10 +13,11 @@ RUN apt-get update && \
     apt-get clean && \
     rm -rf /var/lib/apt/lists/*
 
-ENV MINICONDA_VERSION 23.1.0-1
-ENV PYTHON_VERSION py39
-ENV CONDA_DIR /home/miniconda3
-ENV LATEST_CONDA_SCRIPT "Miniconda3-${PYTHON_VERSION}_${MINICONDA_VERSION}-Linux-x86_64.sh"
+# install miniconda
+ENV MINICONDA_VERSION=25.5.1-0
+ENV PYTHON_VERSION=py312
+ENV CONDA_DIR=/home/miniconda3
+ENV LATEST_CONDA_SCRIPT="Miniconda3-${PYTHON_VERSION}_${MINICONDA_VERSION}-Linux-x86_64.sh"
 
 RUN wget --quiet https://repo.anaconda.com/miniconda/$LATEST_CONDA_SCRIPT -O ~/miniconda.sh && \
     bash ~/miniconda.sh -b -p $CONDA_DIR && \

kilosort-compiled/kilosort2_5-compiled/build.sh

@@ -1,4 +1,4 @@
 #!/bin/bash
 
-docker build -t spikeinterface/kilosort2_5-compiled-base:latest -t spikeinterface/kilosort2_5-compiled-base:0.3.0 .
+docker build -t spikeinterface/kilosort2_5-compiled-base:latest -t spikeinterface/kilosort2_5-compiled-base:2.5.2_cuda-11.2.2 .
 

kilosort-compiled/kilosort3-compiled/Dockerfile

@@ -13,10 +13,11 @@ RUN apt-get update && \
     apt-get clean && \
     rm -rf /var/lib/apt/lists/*
 
-ENV MINICONDA_VERSION 23.1.0-1
-ENV PYTHON_VERSION py39
-ENV CONDA_DIR /home/miniconda3
-ENV LATEST_CONDA_SCRIPT "Miniconda3-${PYTHON_VERSION}_${MINICONDA_VERSION}-Linux-x86_64.sh"
+# install miniconda
+ENV MINICONDA_VERSION=25.5.1-0
+ENV PYTHON_VERSION=py312
+ENV CONDA_DIR=/home/miniconda3
+ENV LATEST_CONDA_SCRIPT="Miniconda3-${PYTHON_VERSION}_${MINICONDA_VERSION}-Linux-x86_64.sh"
 
 RUN wget --quiet https://repo.anaconda.com/miniconda/$LATEST_CONDA_SCRIPT -O ~/miniconda.sh && \
     bash ~/miniconda.sh -b -p $CONDA_DIR && \

kilosort-compiled/kilosort3-compiled/build.sh

@@ -1,3 +1,3 @@
 #!/bin/bash
 
-docker build -t spikeinterface/kilosort3-compiled-base:latest -t spikeinterface/kilosort3-compiled-base:0.3.0 .
+docker build -t spikeinterface/kilosort3-compiled-base:latest -t spikeinterface/kilosort3-compiled-base:3.0.2_cuda-11.2.2 .