|
| 1 | +FROM nvcr.io/nvidia/tensorflow:20.06-tf1-py3 |
| 2 | +# NVIDIA DIGITS runs on Python 3.6 |
| 3 | + |
| 4 | +RUN apt-get update && \ |
| 5 | + apt-get install -y libsm6 libxext6 libxrender-dev |
| 6 | + |
| 7 | +ENV PYTHONUNBUFFERED=1 \ |
| 8 | + LANG=C.UTF-8 |
| 9 | + |
| 10 | +RUN /usr/bin/python3 -m pip install -U pip setuptools && \ |
| 11 | + /usr/bin/python3 -m pip install --no-cache-dir keras && \ |
| 12 | + /usr/bin/python3 -m pip install --no-cache-dir pillow && \ |
| 13 | + /usr/bin/python3 -m pip install --no-cache-dir mpi4py && \ |
| 14 | + /usr/bin/python3 -m pip install --no-cache-dir jupyter |
| 15 | + |
| 16 | +# Install ipython kernelspec |
| 17 | +RUN /usr/bin/python3 -m ipykernel install --display-name "Python 3.6 (NGC 20.06 / TensorFlow 1.15) on Backend.AI" && \ |
| 18 | + cat /usr/local/share/jupyter/kernels/python3/kernel.json |
| 19 | + |
| 20 | +# Backend.AI specifics |
| 21 | +LABEL ai.backend.kernelspec="1" \ |
| 22 | + ai.backend.envs.corecount="OPENBLAS_NUM_THREADS,OMP_NUM_THREADS,NPROC" \ |
| 23 | + ai.backend.features="batch query uid-match user-input" \ |
| 24 | + ai.backend.base-distro="ubuntu16.04" \ |
| 25 | + ai.backend.accelerators="cuda" \ |
| 26 | + ai.backend.resource.min.cpu="1" \ |
| 27 | + ai.backend.resource.min.mem="1g" \ |
| 28 | + ai.backend.resource.min.cuda.device=1 \ |
| 29 | + ai.backend.resource.min.cuda.shares=0.1 \ |
| 30 | + ai.backend.base-distro="ubuntu16.04" \ |
| 31 | + ai.backend.runtime-type="python" \ |
| 32 | + ai.backend.runtime-path="/usr/bin/python" \ |
| 33 | + ai.backend.service-ports="ipython:pty:3000,jupyter:http:8080,jupyterlab:http:8090,tensorboard:http:6006" |
| 34 | + |
| 35 | +# vim: ft=dockerfile |
0 commit comments