FROM nvidia/cuda:11.2.2-cudnn8-runtime-ubuntu20.04

ARG NB_USER="jovyan"

USER root
ENV DEBIAN_FRONTEND=noninteractive
RUN apt-get update && apt-get install -y \
		g++ \
		nodejs \
		python3.8 \
		python3-pip \
		sudo \
		curl \
		&&\
	ln --force --symbolic /usr/bin/python3 /usr/bin/python &&\
	pip3 install \
		jupyterlab \
		pylint \
		tensorflow-gpu==2.7.0 \
		tensorboard==2.7.0 \
		colored \
		matplotlib \
		seaborn \
		scikit-learn \
		scipy \
		tqdm \
		networkx \
		&&\
	curl -fsSL https://deb.nodesource.com/setup_17.x | bash - &&\
	apt-get install -y nodejs &&\
	jupyter labextension install @jupyter-widgets/jupyterlab-manager
RUN useradd \
		--create-home \
		--password "hashlife" \
		--shell /usr/bin/bash \
		--user-group \
		--groups sudo \
		$NB_USER &&\
	echo "$NB_USER:hashlife" | chpasswd  &&\
	mkdir -p /data &&\
	chown $NB_USER /data


COPY --chown=$NB_USER jupyter_notebook_config.py /home/$NB_USER/.jupyter/

USER $NB_USER
WORKDIR /data
#ENV LD_LIBRARY_PATH /usr/local/cuda-10.1/compat:$LD_LIBRARY_PATH
CMD ["jupyter", "lab", "--ContentsManager.allow_hidden=True"]