File tree Expand file tree Collapse file tree 1 file changed +11
-5
lines changed Expand file tree Collapse file tree 1 file changed +11
-5
lines changed Original file line number Diff line number Diff line change @@ -62,6 +62,15 @@ RUN python3 -m pip install --no-cache-dir \
62
62
tensorflow-plot==0.3.2 \
63
63
tensorflow_text==2.4.1 \
64
64
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
65
74
66
75
RUN python3 -m pip install --extra-index-url \
67
76
https://developer.download.nvidia.com/compute/redist nvidia-dali-cuda100 \
@@ -87,11 +96,6 @@ RUN python3 -m pip uninstall -y torch && \
87
96
88
97
RUN python3 -m pip install --no-cache-dir mxnet mkl
89
98
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
-
95
99
# Install Horovod, temporarily using CUDA stubs
96
100
RUN ldconfig /usr/local/cuda/targets/x86_64-linux/lib/stubs && \
97
101
HOROVOD_GPU_ALLREDUCE=NCCL HOROVOD_GPU_BROADCAST=NCCL \
@@ -160,6 +164,8 @@ RUN apt autoclean && \
160
164
rm -rf /root/.cache && \
161
165
rm -rf /tmp/*
162
166
167
+ # Copy Backend.Ai multi-node support
168
+ COPY runner-scripts/bootstrap.sh runner-scripts/setup_multinode.py /opt/container/
163
169
COPY ./service-defs /etc/backend.ai/service-defs
164
170
165
171
# Install ipython kernelspec
You can’t perform that action at this time.
0 commit comments