Skip to content

Commit 8da1095

Browse files
committed
Add iproute2 to Dockerfile
1 parent 31ef077 commit 8da1095

File tree

2 files changed

+6
-0
lines changed

2 files changed

+6
-0
lines changed

g4ad_rocm_build/Dockerfile

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -178,6 +178,9 @@ RUN rm install_ompi.sh
178178

179179
ENV LD_LIBRARY_PATH="/opt/rocm/lib:/opt/rocm/rccl/lib:/usr/lib:/usr/lib/openmpi:/usr/lib/ucc:/usr/lib/ucx"
180180

181+
# Insstall additional utils
182+
RUN apt update && apt install -y iproute2
183+
181184
# Configure PyTorch instalation environment
182185
ENV USE_MPI=1
183186
ENV BUILD_TEST=0

g4dn_cuda_build/Dockerfile

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -112,6 +112,9 @@ RUN if [ -h /usr/local/cuda-11.6/cuda-11.6 ]; then rm /usr/local/cuda-11.6/cuda-
112112
RUN if [ -h /usr/local/cuda-11.7/cuda-11.7 ]; then rm /usr/local/cuda-11.7/cuda-11.7; fi
113113
RUN if [ -h /usr/local/cuda-12.1/cuda-12.1 ]; then rm /usr/local/cuda-12.1/cuda-12.1; fi
114114

115+
# Insstall additional utils
116+
RUN apt update && apt install -y iproute2
117+
115118
# Configure PyTorch instalation environment
116119
ENV TORCH_CUDA_ARCH_LIST="Turing"
117120
ENV USE_MPI=1

0 commit comments

Comments
 (0)