Permalink

	# Copyright (c) Jupyter Development Team.
	# Distributed under the terms of the Modified BSD License.

	FROM base-notebook-gpu

	MAINTAINER Kyle Roth <kylrth@gmail.com>

	USER root

	ARG DEBIAN_FRONTEND=noninteractive

	RUN apt-get update && apt-get install -y --no-install-recommends \
	ant \
	bc \
	build-essential \
	cmake \
	cm-super \
	curl \
	default-jdk \
	doxygen \
	dvipng \
	ffmpeg \
	freeglut3-dev \
	gfortran \
	git \
	gnupg2 \
	g++ \
	inkscape \
	jed \
	less \
	libavcodec-dev \
	libavformat-dev \
	libcupti-dev \
	libcurl3-dev \
	libdc1394-22-dev \
	libeigen3-dev \
	libffi-dev \
	libfreetype6-dev \
	libgl1-mesa-glx \
	libhdf5-dev \
	libjpeg-dev \
	liblapack-dev \
	liblcms2-dev \
	libopenblas-dev \
	libopencore-amrnb-dev \
	libopencore-amrwb-dev \
	libopenexr-dev \
	libpng-dev \
	libsm6 \
	libssl-dev \
	libswscale-dev \
	libtbb-dev \
	libtheora-dev \
	libtiff5-dev \
	libv4l-dev \
	libvips \
	libvips-dev \
	libvips-tools \
	libvorbis-dev \
	libvtk6-dev \
	libwebp-dev \
	libx264-dev \
	libxext-dev \
	libxine2-dev \
	libxrender1 \
	libxvidcore-dev \
	libzmq3-dev \
	lmodern \
	openssh-client \
	pandoc \
	pkg-config \
	qt5-default \
	rsync \
	texlive-latex-extra \
	tmux \
	unzip \
	vim \
	yasm \
	zip \
	zlib1g-dev

	RUN apt-get autoremove -y && \
	apt-get autoclean && \
	apt-get clean && \
	rm -rf /var/lib/apt/lists/*

	ARG NB_USER=deep \
	NB_UID=1000 \
	NB_GID=100
	USER $NB_USER

	# install Go and leave the update-golang.sh command available for the user
	ARG GOVERSION=1.18.2
	RUN git clone https://github.com/udhos/update-golang
	USER root
	RUN cd /home/$NB_USER/update-golang && sed -i 's\|wget -O\|wget --quiet -O\|' update-golang.sh
	# We won't run with sudo here to avoid providing the password, so we'll modify the script so it can
	# run as root.
	RUN cd /home/$NB_USER/update-golang && sed -i 's\|$SUDO_USER\|'"$NB_USER"'\|' update-golang.sh
	RUN cd /home/$NB_USER/update-golang && sed -i 's\|$SUDO_UID\|'"$NB_UID"'\|' update-golang.sh
	RUN cd /home/$NB_USER/update-golang && sed -i 's\|$SUDO_GID\|'"$NB_GID"'\|' update-golang.sh
	RUN cd /home/$NB_USER/update-golang && RELEASE=$GOVERSION ./update-golang.sh -declutter
	USER $NB_USER
	RUN cd /home/$NB_USER/update-golang && git reset --hard

	ENV SHELL=/bin/bash \
	EDITOR=vim

	# rename the base kernel and give it a cool icon
	RUN sed -i 's\|Python 3 (ipykernel)\|base\|' /opt/conda/share/jupyter/kernels/python3/kernel.json
	COPY torch_tf_32.png /opt/conda/share/jupyter/kernels/python3/logo-32x32.png
	COPY torch_tf_64.png /opt/conda/share/jupyter/kernels/python3/logo-64x64.png

	# install Pytorch (force CUDA version even if building with default runtime)
	RUN conda install --yes \
	pytorch=1.11.0=cuda \
	torchvision=0.12.0 \
	torchaudio=0.11.0 \
	cudatoolkit=11.4.1 \
	-c pytorch -c nvidia
	# install TensorFlow
	RUN pip install --no-cache-dir \
	tensorflow-gpu==2.9.1 \
	tensorboard==2.9.0
	# version other important data science packages
	RUN conda install --yes \
	scikit-image=0.19.2 \
	scikit-learn=1.1.1 \
	scipy=1.8.1 \
	matplotlib=3.5.2
	# no need to version these other things, but we will add them after the others so we get the latest
	# when the cache is busted
	RUN conda install --yes \
	black \
	Cython \
	cupy \
	Flask \
	h5py \
	ipykernel \
	ipython \
	nltk \
	nose \
	numpy \
	pandas \
	pandarallel \
	path.py \
	Pillow \
	protobuf \
	pyyaml \
	seaborn \
	sentencepiece \
	setuptools \
	spacy \
	statsmodels \
	tensorboard \
	tqdm \
	transformers \
	wheel

	# # to get the lab extension working with lab v3, we have to install directly from git
	RUN pip install --no-cache-dir git+https://github.com/twalcari/jupyterlab_tensorboard.git

	# We'll install geopandas in a separate env because its dependencies are pretty different
	RUN conda create -n geopandas \
	ipykernel \
	geopandas=0.10.2 \
	pandas \
	pandarallel \
	scikit-image \
	scikit-learn \
	scipy \
	matplotlib
	RUN conda run -n geopandas ipython kernel install --user --name=geopandas
	COPY geopandas_32.png /home/deep/.local/share/jupyter/kernels/geopandas/logo-32x32.png
	COPY geopandas_64.png /home/deep/.local/share/jupyter/kernels/geopandas/logo-64x64.png

	RUN conda build purge-all
	RUN conda init bash

	# import matplotlib the first time to build the font cache
	ENV XDG_CACHE_HOME /home/$NB_USER/.cache/
	RUN MPLBACKEND=Agg python -c "import matplotlib.pyplot"

	# clean up any temporary data
	USER root
	RUN rm -rf /tmp/* /var/tmp/*
	USER $NB_USER

	COPY README.md /home/$NB_USER/README.md

	EXPOSE 5000 6006