jupyter setup

Author: rusenask

Makefile

@@ -1,7 +1,14 @@
-VERSION ?= latest
+VERSION ?= 22.04
 
 slurmdbd:
 	cd docker/slurmdbd && docker build -f Dockerfile --platform linux/amd64 -t ghcr.io/synpse-hq/slurmdbd:${VERSION} .
 
 push-slurmdbd: slurmdbd
-	docker push ghcr.io/synpse-hq/slurmdbd:${VERSION}
+	docker push ghcr.io/synpse-hq/slurmdbd:${VERSION}
+
+
+jupyter:
+	cd docker/jupyter && docker build -f Dockerfile --platform linux/amd64 -t ghcr.io/synpse-hq/slurm-jupyter:${VERSION} .
+
+push-jupyter: jupyter
+	docker push ghcr.io/synpse-hq/slurm-jupyter:${VERSION}

docker/jupyter/Dockerfile

@@ -0,0 +1,28 @@
+FROM ubuntu:22.04
+
+# RUN apt update -y && apt install munge -y && apt install vim -y && apt install build-essential -y && apt install git -y && apt-get install mariadb-server -y && apt install wget -y
+RUN apt update -y && apt install munge vim build-essential git wget -y
+
+ARG DEBIAN_FRONTEND=noninteractive
+
+RUN apt install slurm-client -y
+RUN apt install curl dirmngr apt-transport-https lsb-release ca-certificates -y
+RUN curl -sL https://deb.nodesource.com/setup_12.x | bash -
+RUN apt install sudo -y && apt install python3.9 python3-pip -y && useradd -m admin -s /usr/bin/bash -d /home/admin && echo "admin:admin" | chpasswd && adduser admin sudo && echo "admin     ALL=(ALL) NOPASSWD:ALL" >> /etc/sudoers
+
+RUN apt update -y && apt install nodejs -y && npm install -g configurable-http-proxy && pip3 install jupyterlab==2.1.2 
+
+COPY slurm.conf /etc/slurm-llnl/
+COPY cgroup.conf /etc/slurm-llnl/
+COPY docker-entrypoint.sh /etc/slurm-llnl/
+
+WORKDIR /home/admin
+
+EXPOSE 8888
+
+ENV USER admin
+ENV SHELL bash
+
+RUN apt install libopenmpi-dev -y && pip3 install mpi4py && pip3 install jupyterlab_slurm
+
+ENTRYPOINT ["/etc/slurm-llnl/docker-entrypoint.sh"]

docker/jupyter/cgroup.conf

@@ -0,0 +1,5 @@
+CgroupAutomount=yes
+CgroupReleaseAgentDir="/etc/slurm/cgroup"
+ConstrainCores=yes
+ConstrainDevices=yes
+ConstrainRAMSpace=yes

docker/jupyter/docker-entrypoint.sh

@@ -0,0 +1,7 @@
+#!/bin/bash
+
+sudo systemctl start munge
+
+jupyter lab --no-browser --allow-root --ip=0.0.0.0 --NotebookApp.token='' --NotebookApp.password=''
+
+tail -f /dev/null

docker/jupyter/slurm.conf

@@ -0,0 +1 @@
+#SET BY SYNPSE