Skip to content

Commit eea01a1

Browse files
committed
NGC Tensorflow 18.08 py2
1 parent 9b58180 commit eea01a1

File tree

1 file changed

+41
-0
lines changed

1 file changed

+41
-0
lines changed
Lines changed: 41 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,41 @@
1+
FROM nvcr.io/nvidia/tensorflow:18.08-py2
2+
# NVIDIA DIGITS runs on Python 2.7
3+
4+
RUN apt-get update && \
5+
apt-get install -y libsm6 libxext6 libxrender-dev
6+
7+
ENV PYTHONUNBUFFERED=1 \
8+
PATH=/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:$PATH \
9+
LANG=C.UTF-8
10+
11+
RUN /usr/bin/python -m pip install -U pip setuptools && \
12+
/usr/bin/python -m pip install --no-cache-dir keras && \
13+
/usr/bin/python -m pip install --no-cache-dir keras_applications && \
14+
/usr/bin/python -m pip install --no-cache-dir keras_preprocessing && \
15+
/usr/bin/python -m pip install --no-cache-dir ipython && \
16+
/usr/bin/python -m pip install --no-cache-dir pillow && \
17+
/usr/bin/python -m pip install --no-cache-dir h5py && \
18+
/usr/bin/python -m pip install --no-cache-dir jupyter && \
19+
/usr/bin/python -m pip install --no-cache-dir jupyterlab
20+
21+
# Install ipython kernelspec
22+
RUN /usr/bin/python -m ipykernel install \
23+
--display-name "Python 2.7 (NGC/TensorFlow 18.08) on Backend.AI" && \
24+
cat /usr/local/share/jupyter/kernels/python2/kernel.json
25+
26+
# Backend.AI specifics
27+
COPY policy.yml /etc/backend.ai/jail/policy.yml
28+
LABEL ai.backend.kernelspec="1" \
29+
ai.backend.envs.corecount="OPENBLAS_NUM_THREADS,OMP_NUM_THREADS,NPROC" \
30+
ai.backend.features="batch uid-match" \
31+
ai.backend.accelerators="cuda" \
32+
ai.backend.resource.min.cpu="1" \
33+
ai.backend.resource.min.mem="1g" \
34+
ai.backend.resource.min.cuda.device=1 \
35+
ai.backend.resource.min.cuda.shares=0.1 \
36+
ai.backend.base-distro="ubuntu16.04" \
37+
ai.backend.runtime-type="python" \
38+
ai.backend.runtime-path="/usr/bin/python" \
39+
ai.backend.service-ports="ipython:pty:3000,tensorboard:http:6006,jupyter:http:8080,jupyterlab:http:8090"
40+
41+
# vim: ft=dockerfile

0 commit comments

Comments
 (0)