David Galiffi
88aa2d3cbe
- Renames the CMake option "ROCPROFSYS_USE_HIP" to "ROCPROFSYS_USE_ROCM" - Remove the "ROCPROFSYS_USE_ROCM_SMI option. Controlled with the "ROCPROFSYS_USE_ROCM" option, instead. - Runtime configuration can still toggle ROCPROFSYS_USE_ROCM_SMI to disable the sampling. - Rename ROCPROFSYS_HIP_VERSION macro to ROCPROFSYS_ROCM_VERSION and remove blocks for `ROCPROFSYS_ROCM_VERSION < 60000` - Remove ROCPROFSYS_USE_ROCTRACER and ROCPROFSYS_USE_ROCPROFILER - Update test cases - Update docker files and workflows to install cmake 3.21, which is required for the rocprofiler-sdk findPackage script. - Removed rocm-6.2 from workflows due to a rocprofiler-sdk API change.
59 строки
2.1 KiB
Docker
59 строки
2.1 KiB
Docker
|
|
ARG DISTRO
|
|
ARG VERSION
|
|
FROM ${DISTRO}:${VERSION}
|
|
|
|
ENV HOME /root
|
|
ENV LANG en_US.UTF-8
|
|
ENV LANGUAGE en_US
|
|
ENV LC_ALL C
|
|
ENV SHELL /bin/bash
|
|
ENV BASH_ENV /etc/bash.bashrc
|
|
ENV DEBIAN_FRONTEND noninteractive
|
|
|
|
WORKDIR /tmp
|
|
SHELL [ "/bin/bash", "-c" ]
|
|
|
|
ARG EXTRA_PACKAGES=""
|
|
ARG ROCM_REPO_VERSION="debian"
|
|
ARG ROCM_VERSION="0.0"
|
|
ARG ROCM_REPO_DIST="ubuntu"
|
|
ARG PYTHON_VERSIONS="6 7 8 9 10 11 12"
|
|
ARG AMDGPU_DEB=6.0/ubuntu/focal/amdgpu-install_6.0.60000-1_all.deb
|
|
ENV PATH ${HOME}/.local/bin:${PATH}
|
|
|
|
RUN apt-get update && \
|
|
apt-get dist-upgrade -y && \
|
|
apt-get install -y apt-utils autoconf autotools-dev bash-completion bison \
|
|
build-essential cmake curl flex gettext git-core gnupg2 libnuma1 libopenmpi-dev \
|
|
libpapi-dev libpfm4-dev librpm-dev libtool libudev1 lsb-release m4 \
|
|
python3-pip rpm texinfo wget && \
|
|
OS_VERSION=$(cat /etc/os-release | grep VERSION_ID | sed 's/=/ /'1 | awk '{print $NF}' | sed 's/"//g') && \
|
|
if [ "${OS_VERSION}" == "24.04" ]; then \
|
|
python3 -m pip install --break-system-packages 'cmake==3.21'; \
|
|
else \
|
|
python3 -m pip install 'cmake==3.21'; \
|
|
fi
|
|
|
|
RUN if [ "${ROCM_VERSION}" != "0.0" ]; then \
|
|
wget https://repo.radeon.com/amdgpu-install/${ROCM_VERSION}/ubuntu/${ROCM_REPO_DIST}/${AMDGPU_DEB} && \
|
|
apt-get install -y ./${AMDGPU_DEB} && \
|
|
apt-get update && \
|
|
apt-get install -y rocm-dev rocm-smi-lib roctracer-dev rocprofiler-dev rccl-dev libpciaccess0 ${EXTRA_PACKAGES} && \
|
|
apt-get autoclean; \
|
|
fi
|
|
|
|
RUN wget https://repo.continuum.io/miniconda/Miniconda3-latest-Linux-x86_64.sh -O miniconda.sh && \
|
|
bash miniconda.sh -b -p /opt/conda && \
|
|
export PATH="/opt/conda/bin:${PATH}" && \
|
|
conda config --set always_yes yes --set changeps1 no && \
|
|
conda update -c defaults -n base conda && \
|
|
for i in ${PYTHON_VERSIONS}; do conda create -n py3.${i} -c defaults -c conda-forge python=3.${i} pip; done && \
|
|
for i in ${PYTHON_VERSIONS}; do /opt/conda/envs/py3.${i}/bin/python -m pip install numpy perfetto dataclasses; done && \
|
|
conda clean -a -y && \
|
|
conda init
|
|
|
|
ENV LC_ALL C.UTF-8
|
|
WORKDIR /home
|
|
SHELL [ "/bin/bash", "--login", "-c" ]
|