Skip to content

Commit e039e37

Browse files
committed
move tfd at tensorflow 2.2
1 parent 822764b commit e039e37

File tree

1 file changed

+1
-4
lines changed

1 file changed

+1
-4
lines changed

python-tensorflow/Dockerfile.2.2-py36-cuda10.1

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -65,7 +65,7 @@ RUN python3 -m pip install --no-cache-dir --extra-index-url \
6565
RUN ldconfig /usr/local/cuda/targets/x86_64-linux/lib/stubs && \
6666
HOROVOD_GPU_ALLREDUCE=NCCL HOROVOD_GPU_BROADCAST=NCCL \
6767
HOROVOD_WITH_TENSORFLOW=1 HOROVOD_WITHOUT_PYTORCH=1 HOROVOD_WITHOUT_MXNET=1 \
68-
pip install --no-cache-dir horovod==0.19.4 && \
68+
pip install --no-cache-dir horovod==0.19.5 && \
6969
ldconfig
7070

7171
RUN python3 -m pip install --no-cache-dir \
@@ -77,9 +77,6 @@ RUN python3 -m pip install --no-cache-dir \
7777
Run python3 -m ipykernel install --display-name "TensorFlow 2.2 on Python 3.6 (CUDA 10.1)" && \
7878
cat /usr/local/share/jupyter/kernels/python3/kernel.json
7979

80-
RUN jupyter nbextension enable --py --sys-prefix widgetsnbextension && \
81-
jupyter nbextension enable --py tensorflow_model_analysis
82-
8380
# Backend.AI specifics
8481
LABEL ai.backend.kernelspec="1" \
8582
ai.backend.envs.corecount="OPENBLAS_NUM_THREADS,OMP_NUM_THREADS,NPROC" \

0 commit comments

Comments
 (0)