Skip to content

Commit cd4aa23

Browse files
committed
add renner-scripts
1 parent 3e9514e commit cd4aa23

File tree

1 file changed

+11
-5
lines changed

1 file changed

+11
-5
lines changed

python-ff/Dockerfile.21.01-py38-cuda11.1

Lines changed: 11 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -62,6 +62,15 @@ RUN python3 -m pip install --no-cache-dir \
6262
tensorflow-plot==0.3.2 \
6363
tensorflow_text==2.4.1 \
6464
tensorflow-gan==2.0.0
65+
RUN python3 -m pip uninstall --no-cache-dir -y \
66+
tensorflow==2.3.0 \
67+
tensorflow-gpu==2.4.1 \
68+
tensorboard==2.3.0 \
69+
tensorboard-plugin-wit==1.7.0 && \
70+
python3 -m pip install --no-cache-dir \
71+
tensorflow-gpu==2.4.1 \
72+
tensorboard-plugin-wit==1.8.0 \
73+
tensorboard==2.4.1
6574

6675
RUN python3 -m pip install --extra-index-url \
6776
https://developer.download.nvidia.com/compute/redist nvidia-dali-cuda100 \
@@ -87,11 +96,6 @@ RUN python3 -m pip uninstall -y torch && \
8796

8897
RUN python3 -m pip install --no-cache-dir mxnet mkl
8998

90-
#RUN python3 -m pip uninstall -y tensorflow && \
91-
# python3 -m pip install --no-cache-dir \
92-
# tensorflow-gpu==2.4.0 \
93-
# tensorboard==2.4.0
94-
9599
# Install Horovod, temporarily using CUDA stubs
96100
RUN ldconfig /usr/local/cuda/targets/x86_64-linux/lib/stubs && \
97101
HOROVOD_GPU_ALLREDUCE=NCCL HOROVOD_GPU_BROADCAST=NCCL \
@@ -160,6 +164,8 @@ RUN apt autoclean && \
160164
rm -rf /root/.cache && \
161165
rm -rf /tmp/*
162166

167+
# Copy Backend.Ai multi-node support
168+
COPY runner-scripts/bootstrap.sh runner-scripts/setup_multinode.py /opt/container/
163169
COPY ./service-defs /etc/backend.ai/service-defs
164170

165171
# Install ipython kernelspec

0 commit comments

Comments
 (0)