Skip to content

Commit 2dd188e

Browse files
committed
remove centos.stream dockerfile
1 parent 0a6e1d6 commit 2dd188e

File tree

1 file changed

+21
-13
lines changed

1 file changed

+21
-13
lines changed

.ci/docker/centos-rocm/Dockerfile

+21-13
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,7 @@
11
ARG CENTOS_VERSION
22

3-
FROM centos:${CENTOS_VERSION}
3+
FROM quay.io/centos/centos:stream${CENTOS_VERSION}
44

5-
ARG CENTOS_VERSION
65

76
# Set AMD gpu targets to build for
87
ARG PYTORCH_ROCM_ARCH
@@ -11,23 +10,25 @@ ENV PYTORCH_ROCM_ARCH ${PYTORCH_ROCM_ARCH}
1110
# Install required packages to build Caffe2
1211

1312
# Install common dependencies (so that this step can be cached separately)
13+
ARG EC2
1414
COPY ./common/install_base.sh install_base.sh
1515
RUN bash ./install_base.sh && rm install_base.sh
1616

17+
#Install langpack
18+
RUN yum install -y glibc-langpack-en
19+
1720
# Update CentOS git version
1821
RUN yum -y remove git
1922
RUN yum -y remove git-*
20-
RUN yum -y install https://packages.endpoint.com/rhel/7/os/x86_64/endpoint-repo-1.9-1.x86_64.rpm || \
21-
(yum -y install https://packages.endpointdev.com/rhel/7/os/x86_64/endpoint-repo-1.9-1.x86_64.rpm && \
22-
sed -i "s/packages.endpoint/packages.endpointdev/" /etc/yum.repos.d/endpoint.repo)
2323
RUN yum install -y git
2424

2525
# Install devtoolset
26-
ARG DEVTOOLSET_VERSION
27-
COPY ./common/install_devtoolset.sh install_devtoolset.sh
28-
RUN bash ./install_devtoolset.sh && rm install_devtoolset.sh
26+
RUN dnf install -y rpmdevtools
2927
ENV BASH_ENV "/etc/profile"
3028

29+
# Install ninja
30+
RUN dnf --enablerepo=crb install -y ninja-build
31+
3132
# (optional) Install non-default glibc version
3233
ARG GLIBC_VERSION
3334
COPY ./common/install_glibc.sh install_glibc.sh
@@ -41,6 +42,7 @@ RUN bash ./install_user.sh && rm install_user.sh
4142
# Install conda and other packages (e.g., numpy, pytest)
4243
ARG ANACONDA_PYTHON_VERSION
4344
ARG CONDA_CMAKE
45+
ARG BUILD_ENVIRONMENT
4446
ENV ANACONDA_PYTHON_VERSION=$ANACONDA_PYTHON_VERSION
4547
ENV PATH /opt/conda/envs/py_$ANACONDA_PYTHON_VERSION/bin:/opt/conda/bin:$PATH
4648
COPY requirements-ci.txt /opt/conda/requirements-ci.txt
@@ -55,7 +57,14 @@ RUN if [ -n "${PROTOBUF}" ]; then bash ./install_protobuf.sh; fi
5557
RUN rm install_protobuf.sh
5658
ENV INSTALLED_PROTOBUF ${PROTOBUF}
5759

58-
# (optional) Install vision packages like OpenCV
60+
# (optional) Install database packages like LMDB and LevelDB
61+
ARG DB
62+
COPY ./common/install_db.sh install_db.sh
63+
RUN if [ -n "${DB}" ]; then bash ./install_db.sh; fi
64+
RUN rm install_db.sh
65+
ENV INSTALLED_DB ${DB}
66+
67+
# (optional) Install vision packages like OpenCV and ffmpeg
5968
ARG VISION
6069
COPY ./common/install_vision.sh ./common/cache_vision_models.sh ./common/common_utils.sh ./
6170
RUN if [ -n "${VISION}" ]; then bash ./install_vision.sh; fi
@@ -68,11 +77,10 @@ COPY ./common/install_rocm.sh install_rocm.sh
6877
RUN bash ./install_rocm.sh
6978
RUN rm install_rocm.sh
7079
COPY ./common/install_rocm_magma.sh install_rocm_magma.sh
71-
RUN bash ./install_rocm_magma.sh ${ROCM_VERSION}
80+
RUN bash ./install_rocm_magma.sh
7281
RUN rm install_rocm_magma.sh
73-
COPY ./common/install_amdsmi.sh install_amdsmi.sh
74-
RUN bash ./install_amdsmi.sh
75-
RUN rm install_amdsmi.sh
82+
83+
ENV ROCM_PATH /opt/rocm
7684
ENV PATH /opt/rocm/bin:$PATH
7785
ENV PATH /opt/rocm/hcc/bin:$PATH
7886
ENV PATH /opt/rocm/hip/bin:$PATH

0 commit comments

Comments
 (0)