Skip to content

Commit 2666307

Browse files
committed
update: add multi-node support for Backend.AI
1 parent ba1da84 commit 2666307

File tree

3 files changed

+9
-0
lines changed

3 files changed

+9
-0
lines changed

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

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -40,6 +40,9 @@ COPY ./service-defs /etc/backend.ai/service-defs
4040
Run python3 -m ipykernel install --display-name "TensorFlow 2.3 on Python 3.6 & CUDA 10.1" && \
4141
cat /usr/local/share/jupyter/kernels/python3/kernel.json
4242

43+
# Copy Backend.Ai multi-node support
44+
COPY runner-scripts/bootstrap.sh runner-scripts/setup_multinode.py /opt/container/
45+
4346
# Backend.AI specifics
4447
LABEL ai.backend.kernelspec="1" \
4548
ai.backend.envs.corecount="OPENBLAS_NUM_THREADS,OMP_NUM_THREADS,NPROC" \

python-tensorflow/Dockerfile.2.4-py36-cuda11.1

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -77,6 +77,9 @@ RUN python3 -m pip install --no-cache-dir \
7777
Run python3 -m ipykernel install --display-name "TensorFlow 2.4 on Python 3.6 & CUDA 11ᇂᇂᇂᇂ.1" && \
7878
cat /usr/local/share/jupyter/kernels/python3/kernel.json
7979

80+
# Copy Backend.Ai multi-node support
81+
COPY runner-scripts/bootstrap.sh runner-scripts/setup_multinode.py /opt/container/
82+
8083
# Backend.AI specifics
8184
LABEL ai.backend.kernelspec="1" \
8285
ai.backend.envs.corecount="OPENBLAS_NUM_THREADS,OMP_NUM_THREADS,NPROC" \

python-tensorflow/Dockerfile.2.4-py38-cuda11.1

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -96,6 +96,9 @@ COPY ./service-defs /etc/backend.ai/service-defs
9696
Run python3 -m ipykernel install --display-name "TensorFlow 2.4 on Python 3.8 & CUDA 11.1" && \
9797
cat /usr/local/share/jupyter/kernels/python3/kernel.json
9898

99+
# Copy Backend.Ai multi-node support
100+
COPY runner-scripts/bootstrap.sh runner-scripts/setup_multinode.py /opt/container/
101+
99102
# Backend.AI specifics
100103
LABEL ai.backend.kernelspec="1" \
101104
ai.backend.envs.corecount="OPENBLAS_NUM_THREADS,OMP_NUM_THREADS,NPROC" \

0 commit comments

Comments
 (0)