Dockerfile - Upgrade mlc to v3.11 (#620)
**Description** Upgrade mlc to v3.11.
This commit is contained in:
Родитель
cc89ee591c
Коммит
dc3846cbd4
|
@ -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
|
||||
|
|
Загрузка…
Ссылка в новой задаче