1
1
FROM lablup/common-base:20.07-py36-cuda10.1
2
2
3
3
# Install TensorFlow, Keras, PyTorch and MXNet
4
- ENV TENSORFLOW_VERSION=2.3.0-rc2
4
+ ENV TENSORFLOW_VERSION=2.3.0
5
5
6
6
ENV PYTHONUNBUFFERED=1 \
7
7
LD_LIBRARY_PATH="/usr/local/cuda/extras/CUPTI/lib64:/usr/local/cuda/lib64:/usr/local/nvidia/lib64:/usr/include/x86_64-linux-gnu" \
@@ -58,11 +58,17 @@ RUN python3 -m pip install --no-cache-dir \
58
58
RUN python3 -m pip install --no-cache-dir \
59
59
tensorflow==${TENSORFLOW_VERSION}
60
60
61
+ RUN python3 -m pip install --no-cache-dir --extra-index-url \
62
+ https://developer.download.nvidia.com/compute/redist \
63
+ nvidia-dali-cuda100 && \
64
+ python3 -m pip install --no-cache-dir --extra-index-url \
65
+ https://developer.download.nvidia.com/compute/redist nvidia-dali-tf-plugin-cuda100
66
+
61
67
# Install Horovod, temporarily using CUDA stubs
62
68
RUN ldconfig /usr/local/cuda/targets/x86_64-linux/lib/stubs && \
63
69
HOROVOD_GPU_ALLREDUCE=NCCL HOROVOD_GPU_BROADCAST=NCCL \
64
70
HOROVOD_WITH_TENSORFLOW=1 HOROVOD_WITHOUT_PYTORCH=1 HOROVOD_WITHOUT_MXNET=1 \
65
- pip install --no-cache-dir horovod==0.19.4 && \
71
+ pip install --no-cache-dir horovod==0.19.5 && \
66
72
ldconfig
67
73
68
74
RUN python3 -m pip install --no-cache-dir \
@@ -74,9 +80,6 @@ RUN python3 -m pip install --no-cache-dir \
74
80
Run python3 -m ipykernel install --display-name "TensorFlow 2.3 on Python 3.6 (CUDA 10.1)" && \
75
81
cat /usr/local/share/jupyter/kernels/python3/kernel.json
76
82
77
- RUN jupyter nbextension enable --py --sys-prefix widgetsnbextension && \
78
- jupyter nbextension enable --py tensorflow_model_analysis
79
-
80
83
# Backend.AI specifics
81
84
LABEL ai.backend.kernelspec="1" \
82
85
ai.backend.envs.corecount="OPENBLAS_NUM_THREADS,OMP_NUM_THREADS,NPROC" \
0 commit comments