File tree Expand file tree Collapse file tree 2 files changed +6
-0
lines changed
Expand file tree Collapse file tree 2 files changed +6
-0
lines changed Original file line number Diff line number Diff line change @@ -178,6 +178,9 @@ RUN rm install_ompi.sh
178178
179179ENV 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
182185ENV USE_MPI=1
183186ENV BUILD_TEST=0
Original file line number Diff line number Diff 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-
112112RUN if [ -h /usr/local/cuda-11.7/cuda-11.7 ]; then rm /usr/local/cuda-11.7/cuda-11.7; fi
113113RUN 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
116119ENV TORCH_CUDA_ARCH_LIST="Turing"
117120ENV USE_MPI=1
You can’t perform that action at this time.
0 commit comments