Skip to content

Commit 747637c

Browse files
committed
NGC tensorflow 20.06, TF1
1 parent 0aa59ae commit 747637c

File tree

1 file changed

+35
-0
lines changed

1 file changed

+35
-0
lines changed
Lines changed: 35 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,35 @@
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

Comments
 (0)