Unverified Commit a7f508e4 authored by guoshzhao's avatar guoshzhao Committed by GitHub
Browse files

Dockerfile: Add dockerfile - Add rocm 4.0 and 4.2 dockerfile with pytorch1.7.0 (#164)

**Description**
Add dockerfile `rocm4.0-pytorch1.7.0.dockerfile` and `rocm4.2-pytorch1.7.0.dockerfile` for `rocm` platform.
parent c8357f4e
FROM rocm/pytorch:rocm4.0_ubuntu18.04_py3.6_pytorch_1.7.0
# OS:
# - Ubuntu: 18.04
# AMD:
# - ROCm: 4.0
# - HIP: 3.21.2
# - MIOpen: 2.9.0
# - RCCL: 2.7.8
# Mellanox:
# - OFED: 5.2-2.2.3.0
LABEL maintainer="SuperBench"
ENV DEBIAN_FRONTEND=noninteractive
RUN wget -qO - http://repo.radeon.com/rocm/apt/debian/rocm.gpg.key | APT_KEY_DONT_WARN_ON_DANGEROUS_USAGE=1 apt-key add - && \
apt-get update && \
apt-get install -y --no-install-recommends \
autoconf \
automake \
build-essential \
curl \
dmidecode \
git \
jq \
libaio-dev \
libcap2 \
libpci-dev \
libtinfo5 \
libtool \
lshw \
net-tools \
openssh-client \
openssh-server \
pciutils \
util-linux \
vim \
wget \
&& \
apt-get autoremove && \
apt-get clean && \
rm -rf /var/lib/apt/lists/* /tmp/*
# Configure SSH
RUN mkdir -p /root/.ssh && \
touch /root/.ssh/authorized_keys && \
mkdir -p /var/run/sshd && \
sed -i "s/[# ]*PermitRootLogin prohibit-password/PermitRootLogin yes/" /etc/ssh/sshd_config && \
sed -i "s/[# ]*Port.*/Port 22/" /etc/ssh/sshd_config && \
echo "PermitUserEnvironment yes" >> /etc/ssh/sshd_config
# Install OFED
ENV OFED_VERSION=5.2-2.2.3.0
RUN cd /tmp && \
wget -q http://content.mellanox.com/ofed/MLNX_OFED-${OFED_VERSION}/MLNX_OFED_LINUX-${OFED_VERSION}-ubuntu18.04-x86_64.tgz && \
tar xzf MLNX_OFED_LINUX-${OFED_VERSION}-ubuntu18.04-x86_64.tgz && \
PATH=/usr/bin:${PATH} MLNX_OFED_LINUX-${OFED_VERSION}-ubuntu18.04-x86_64/mlnxofedinstall --user-space-only --without-fw-update --force --all && \
rm -rf MLNX_OFED_LINUX-${OFED_VERSION}*
ENV PATH="${PATH}" \
LD_LIBRARY_PATH="/usr/local/lib:${LD_LIBRARY_PATH}" \
SB_HOME="/opt/superbench" \
SB_MICRO_PATH="/opt/superbench"
WORKDIR ${SB_HOME}
ADD third_party third_party
RUN ROCM_VERSION=rocm-4.0.0 make -j -C third_party rocm
# Workaround for image having package installed in user path
RUN mv /root/.local/bin/* /opt/conda/bin/ && \
mv /root/.local/lib/python3.6/site-packages/* /opt/conda/lib/python3.6/ && \
rm -rf /root/.local
ADD . .
RUN python3 -m pip install .[torch] && \
make cppbuild
FROM rocm/pytorch:rocm4.2_ubuntu18.04_py3.6_pytorch_1.7.0
# OS:
# - Ubuntu: 18.04
# AMD:
# - ROCm: 4.2
# - HIP: 3.27.5
# - MIOpen: 2.11.0
# - RCCL: 2.8.4
# Mellanox:
# - OFED: 5.2-2.2.3.0
LABEL maintainer="SuperBench"
ENV DEBIAN_FRONTEND=noninteractive
RUN wget -qO - http://repo.radeon.com/rocm/apt/debian/rocm.gpg.key | APT_KEY_DONT_WARN_ON_DANGEROUS_USAGE=1 apt-key add - && \
apt-get update && \
apt-get install -y --no-install-recommends \
autoconf \
automake \
build-essential \
curl \
dmidecode \
git \
jq \
libaio-dev \
libcap2 \
libpci-dev \
libtinfo5 \
libtool \
lshw \
net-tools \
openssh-client \
openssh-server \
pciutils \
util-linux \
vim \
wget \
&& \
apt-get autoremove && \
apt-get clean && \
rm -rf /var/lib/apt/lists/* /tmp/*
# Configure SSH
RUN mkdir -p /root/.ssh && \
touch /root/.ssh/authorized_keys && \
mkdir -p /var/run/sshd && \
sed -i "s/[# ]*PermitRootLogin prohibit-password/PermitRootLogin yes/" /etc/ssh/sshd_config && \
sed -i "s/[# ]*Port.*/Port 22/" /etc/ssh/sshd_config && \
echo "PermitUserEnvironment yes" >> /etc/ssh/sshd_config
# Install OFED
ENV OFED_VERSION=5.2-2.2.3.0
RUN cd /tmp && \
wget -q http://content.mellanox.com/ofed/MLNX_OFED-${OFED_VERSION}/MLNX_OFED_LINUX-${OFED_VERSION}-ubuntu18.04-x86_64.tgz && \
tar xzf MLNX_OFED_LINUX-${OFED_VERSION}-ubuntu18.04-x86_64.tgz && \
PATH=/usr/bin:${PATH} MLNX_OFED_LINUX-${OFED_VERSION}-ubuntu18.04-x86_64/mlnxofedinstall --user-space-only --without-fw-update --force --all && \
rm -rf MLNX_OFED_LINUX-${OFED_VERSION}*
ENV PATH="${PATH}" \
LD_LIBRARY_PATH="/usr/local/lib:${LD_LIBRARY_PATH}" \
SB_HOME="/opt/superbench" \
SB_MICRO_PATH="/opt/superbench"
WORKDIR ${SB_HOME}
ADD third_party third_party
RUN ROCM_VERSION=rocm-4.2.0 make -j -C third_party rocm
ADD . .
RUN python3 -m pip install .[torch] && \
make cppbuild
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment