Permalink
191 lines (173 sloc)
4.67 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 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 |