Dockerfile

# 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