Dockerfile - Upgrade mlc to v3.11 (#620)

**Description**
Upgrade mlc to v3.11.
This commit is contained in:
Yuting Jiang 2024-04-18 10:59:36 +08:00 коммит произвёл GitHub
Родитель cc89ee591c
Коммит dc3846cbd4
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: B5690EEEBB952194
6 изменённых файлов: 13 добавлений и 13 удалений

Просмотреть файл

@ -13,7 +13,7 @@ FROM nvcr.io/nvidia/pytorch:20.12-py3
# - HPC-X: v2.8.3
# - NCCL RDMA SHARP plugins: 7cccbc1
# Intel:
# - mlc: v3.10
# - mlc: v3.11
LABEL maintainer="SuperBench"
@ -112,7 +112,7 @@ RUN cd /tmp && \
# Install Intel MLC
RUN cd /tmp && \
wget -q https://downloadmirror.intel.com/763324/mlc_v3.10.tgz -O mlc.tgz && \
wget -q https://downloadmirror.intel.com/793041/mlc_v3.11.tgz -O mlc.tgz && \
tar xzf mlc.tgz Linux/mlc && \
cp ./Linux/mlc /usr/local/bin/ && \
rm -rf ./Linux mlc.tgz

Просмотреть файл

@ -12,7 +12,7 @@ FROM nvcr.io/nvidia/pytorch:23.10-py3
# - OFED: 23.07-0.5.1.2
# - HPC-X: v2.16
# Intel:
# - mlc: v3.10
# - mlc: v3.11
LABEL maintainer="SuperBench"
@ -95,7 +95,7 @@ RUN cd /opt && \
# Install Intel MLC
RUN cd /tmp && \
wget -q https://downloadmirror.intel.com/763324/mlc_v3.10.tgz -O mlc.tgz && \
wget -q https://downloadmirror.intel.com/793041/mlc_v3.11.tgz -O mlc.tgz && \
tar xzf mlc.tgz Linux/mlc && \
cp ./Linux/mlc /usr/local/bin/ && \
rm -rf ./Linux mlc.tgz

Просмотреть файл

@ -17,7 +17,7 @@ FROM ${BASE_IMAGE}
# Mellanox:
# - OFED: 5.2-2.2.3.0
# Intel:
# - mlc: v3.10
# - mlc: v3.11
LABEL maintainer="SuperBench"
@ -98,7 +98,7 @@ RUN cd /tmp && \
# Install Intel MLC
RUN cd /tmp && \
wget -q https://downloadmirror.intel.com/763324/mlc_v3.10.tgz -O mlc.tgz && \
wget -q https://downloadmirror.intel.com/793041/mlc_v3.11.tgz -O mlc.tgz && \
tar xzf mlc.tgz Linux/mlc && \
cp ./Linux/mlc /usr/local/bin/ && \
rm -rf ./Linux mlc.tgz

Просмотреть файл

@ -16,7 +16,7 @@ FROM ${BASE_IMAGE}
# Mellanox:
# - OFED: 5.2-2.2.3.0
# Intel:
# - mlc: v3.10
# - mlc: v3.11
LABEL maintainer="SuperBench"
@ -110,7 +110,7 @@ RUN cd /tmp && \
# Install Intel MLC
RUN cd /tmp && \
wget -q https://downloadmirror.intel.com/763324/mlc_v3.10.tgz -O mlc.tgz && \
wget -q https://downloadmirror.intel.com/793041/mlc_v3.11.tgz -O mlc.tgz && \
tar xzf mlc.tgz Linux/mlc && \
cp ./Linux/mlc /usr/local/bin/ && \
rm -rf ./Linux mlc.tgz
@ -142,7 +142,7 @@ RUN echo PATH="$PATH" > /etc/environment && \
WORKDIR ${SB_HOME}
ADD third_party third_party
RUN make ROCBLAS_BRANCH=release/rocm-rel-5.1 -C third_party rocm -o rocm_hipblaslt -o megatron_deepspeed -o megatron_lm
RUN make ROCBLAS_BRANCH=release/rocm-rel-5.1 -C third_party rocm -o rocm_hipblaslt -o megatron_deepspeed -o megatron_lm
ADD . .
RUN python3 -m pip install --no-cache-dir .[amdworker] && \

Просмотреть файл

@ -8,7 +8,7 @@ FROM ${BASE_IMAGE}
# ROCm:
# - ROCm: 5.7
# Intel:
# - mlc: v3.10
# - mlc: v3.11
LABEL maintainer="SuperBench"
@ -126,7 +126,7 @@ RUN cd /tmp && \
# Install Intel MLC
RUN cd /tmp && \
wget -q https://downloadmirror.intel.com/763324/mlc_v3.10.tgz -O mlc.tgz && \
wget -q https://downloadmirror.intel.com/793041/mlc_v3.11.tgz -O mlc.tgz && \
tar xzf mlc.tgz Linux/mlc && \
cp ./Linux/mlc /usr/local/bin/ && \
rm -rf ./Linux mlc.tgz

Просмотреть файл

@ -14,7 +14,7 @@ FROM ${BASE_IMAGE}
# - openmpi: 4.1.x
# - apex: 1.0.0
# Intel:
# - mlc: v3.10
# - mlc: v3.11
LABEL maintainer="SuperBench"
@ -132,7 +132,7 @@ RUN cd /tmp && \
# Install Intel MLC
RUN cd /tmp && \
wget -q https://downloadmirror.intel.com/763324/mlc_v3.10.tgz -O mlc.tgz && \
wget -q https://downloadmirror.intel.com/793041/mlc_v3.11.tgz -O mlc.tgz && \
tar xzf mlc.tgz Linux/mlc && \
cp ./Linux/mlc /usr/local/bin/ && \
rm -rf ./Linux mlc.tgz