Permalink
105 lines (89 sloc)
3.4 KB
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| # Copyright (c) Jupyter Development Team. | |
| # Distributed under the terms of the Modified BSD License. | |
| FROM nvidia/cuda:11.4.2-cudnn8-devel-ubuntu20.04 | |
| MAINTAINER Kyle Roth <kylrth@gmail.com> | |
| USER root | |
| ARG DEBIAN_FRONTEND=noninteractive | |
| # Nvidia has rotated their GPG keys since this image was created. | |
| RUN apt-key del 7fa2af80 && \ | |
| apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/3bf863cc.pub | |
| RUN apt-get update && \ | |
| apt-get install -yq software-properties-common | |
| RUN apt-get update && apt-get -yq dist-upgrade | |
| RUN apt-get install -yq --no-install-recommends \ | |
| bzip2 \ | |
| ca-certificates \ | |
| curl \ | |
| fonts-liberation \ | |
| locales \ | |
| man \ | |
| manpages-posix \ | |
| sudo \ | |
| wget | |
| RUN yes | unminimize | |
| RUN echo "en_US.UTF-8 UTF-8" > /etc/locale.gen && locale-gen | |
| ENV LC_ALL=en_US.UTF-8 \ | |
| LANG=en_US.UTF-8 \ | |
| LANGUAGE=en_US.UTF-8 | |
| # Install Tini | |
| ARG TINI_VERSION=0.19.0 | |
| ARG TINI_SHA256=93dcc18adc78c65a028a84799ecf8ad40c936fdfc5f2a57b1acda5a8117fa82c | |
| RUN wget --quiet https://github.com/krallin/tini/releases/download/v${TINI_VERSION}/tini && \ | |
| echo "${TINI_SHA256} *tini" | sha256sum -c - && \ | |
| mv tini /usr/local/bin/tini && \ | |
| chmod +x /usr/local/bin/tini | |
| COPY fix-permissions /usr/local/bin/fix-permissions | |
| COPY start.sh /usr/local/bin/ | |
| COPY start-notebook.sh /usr/local/bin/ | |
| COPY start-singleuser.sh /usr/local/bin/ | |
| # create deep user with UID=1000 and in the 'users' group | |
| # and make sure these dirs are writable by the `users` group | |
| ARG CONDA_DIR=/opt/conda \ | |
| SHELL=/bin/bash \ | |
| NB_USER=deep \ | |
| NB_UID=1000 \ | |
| NB_GID=100 | |
| RUN useradd -m -s /bin/bash -N -u $NB_UID $NB_USER && \ | |
| mkdir -p $CONDA_DIR && \ | |
| chown $NB_USER:$NB_GID $CONDA_DIR && \ | |
| fix-permissions /home/$NB_USER && \ | |
| fix-permissions $CONDA_DIR | |
| RUN rm /usr/local/bin/fix-permissions | |
| RUN usermod -aG sudo $NB_USER | |
| RUN echo "deep:changeme" | chpasswd | |
| USER $NB_USER | |
| # configure environment for CUDA | |
| ENV CUDA_HOME /usr/local/cuda-11.4 | |
| ENV LD_LIBRARY_PATH /usr/local/cuda-11.4:/usr/local/cuda-11.4/lib64:$LD_LIBRARY_PATH | |
| # CUPTI libs were moved to /usr/local/cuda/lib64 | |
| ENV LD_LIBRARY_PATH /usr/local/cuda/lib64:$LD_LIBRARY_PATH | |
| ENV LIBRARY_PATH $LD_LIBRARY_PATH | |
| # install conda as non-root user | |
| ARG MINICONDA_VERSION=py39_4.12.0 | |
| ARG MINICONDA_SUM=7843dd7d0a2c53b0df37ca8189672992 | |
| ENV PATH=$CONDA_DIR/bin:$PATH | |
| RUN cd /tmp && \ | |
| wget --quiet https://repo.anaconda.com/miniconda/Miniconda3-${MINICONDA_VERSION}-Linux-x86_64.sh && \ | |
| echo "${MINICONDA_SUM} *Miniconda3-${MINICONDA_VERSION}-Linux-x86_64.sh" | md5sum -c - && \ | |
| /bin/bash Miniconda3-${MINICONDA_VERSION}-Linux-x86_64.sh -f -b -p $CONDA_DIR && \ | |
| rm Miniconda3-${MINICONDA_VERSION}-Linux-x86_64.sh && \ | |
| conda config --system --prepend channels conda-forge && \ | |
| conda config --system --set auto_update_conda false && \ | |
| conda config --system --set show_channel_urls true && \ | |
| conda update --all --quiet --yes && \ | |
| conda install --yes conda-build && conda build purge-all | |
| # install necessary Jupyter components | |
| RUN conda install -y \ | |
| python \ | |
| notebook=6.4.11 \ | |
| jupyterhub=2.3.0 \ | |
| jupyterlab=3.4.2 \ | |
| jupyter \ | |
| six | |
| # create the .jupyter/lab dir so that when we mount user settings it doesn't create this dir as root | |
| RUN mkdir -p /home/$NB_USER/.jupyter/lab | |
| # configure container startup | |
| ENTRYPOINT ["tini", "--"] | |
| CMD ["start-notebook.sh"] | |
| EXPOSE 8888 | |
| WORKDIR /home/$NB_USER |