Merge pull request #70 from SpikeInterface/kilosort4

Kilosort4 image

Author: alejoe91

kilosort4/Dockerfile

@@ -0,0 +1,38 @@
+FROM nvidia/cuda:12.0.0-base-ubuntu20.04
+
+LABEL maintainer="Alessio Buccino <alessiop.buccino@gmail.com>"
+
+# USER root
+# Ubuntu package installs
+RUN apt update && \
+    apt install -y --no-install-recommends \
+    libfftw3-dev \
+    git \
+    wget && \
+    apt clean && \
+    rm -rf /var/lib/apt/lists/*
+
+# install miniconda
+ENV MINICONDA_VERSION 24.1.2-0
+ENV PY_VERSION py311
+ENV CONDA_DIR /home/miniconda3
+ENV LATEST_CONDA_SCRIPT "Miniconda3-${PY_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 && \
+    rm ~/miniconda.sh
+ENV PATH=$CONDA_DIR/bin:$PATH
+RUN conda update conda && \
+    conda install conda-build
+
+# make conda activate command available from /bin/bash --login shells
+RUN echo ". $CONDA_DIR/etc/profile.d/conda.sh" >> /root/.profile 
+# make conda activate command available from /bin/bash --interactive shells
+RUN conda init bash
+
+# install torch
+# RUN conda install -y pytorch pytorch-cuda=11.8 -c pytorch -c nvidia
+RUN pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
+
+# install kilosort4
+RUN pip install kilosort==4.0

kilosort4/build.sh

@@ -0,0 +1,3 @@
+#!/bin/bash
+
+docker build -t spikeinterface/kilosort4-base:latest -t spikeinterface/kilosort4-base:4.0_cuda-12.0.0 .

kilosort4/push.sh

@@ -0,0 +1,3 @@
+#!/bin/bash
+
+docker push --all-tags spikeinterface/kilosort4-base