Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 1 addition & 5 deletions cmd/runtimes/deepspeed/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -3,14 +3,10 @@ FROM nvidia/cuda:13.1.1-devel-ubuntu22.04

# Install libraries required for OpenMPI to work. Image installs OpenMPI 5.0.7
RUN apt update && apt install -y --no-install-recommends \
openssh-server openssh-client libcap2-bin \
openssh-server openssh-client \
g++ libopenmpi-dev \
python3-dev pip && rm -f /usr/bin/python && ln -s /usr/bin/python3 /usr/bin/python && rm -rf /var/lib/apt/lists/*

# Add capability to run sshd as non-root.
RUN setcap CAP_NET_BIND_SERVICE=+eip /usr/sbin/sshd
RUN apt remove libcap2-bin -y

# Configure mpiuser and home directory.
RUN useradd -m mpiuser
WORKDIR /home/mpiuser
Expand Down
6 changes: 1 addition & 5 deletions cmd/runtimes/mlx/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -3,14 +3,10 @@ FROM nvidia/cuda:13.1.1-devel-ubuntu22.04

# Install libraries required for OpenMPI to work. Image installs OpenMPI 5.0.7
RUN apt update && apt install -y --no-install-recommends \
openssh-server openssh-client libcap2-bin \
openssh-server openssh-client \
g++ libopenmpi-dev libblas-dev liblapack-dev liblapacke-dev \
python3-dev pip && rm -f /usr/bin/python && ln -s /usr/bin/python3 /usr/bin/python && rm -rf /var/lib/apt/lists/*

# Add capability to run sshd as non-root.
RUN setcap CAP_NET_BIND_SERVICE=+eip /usr/sbin/sshd
RUN apt remove libcap2-bin -y

# Configure mpiuser and home directory.
RUN useradd -m mpiuser
WORKDIR /home/mpiuser
Expand Down
Loading