Skip to content

Commit e8cf4a6

Browse files
committed
docker: add xpu-smi to intel image for device detection
1 parent 41c79f0 commit e8cf4a6

File tree

5 files changed

+15
-6
lines changed

5 files changed

+15
-6
lines changed

.github/workflows/docker-bases.yml

+1-1
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@ on:
1717
description: 'Build compiler bases'
1818
schedule:
1919
# Run once a week
20-
- cron: "0 13 * * 1"
20+
- cron: "0 0 1 * *"
2121

2222
jobs:
2323
#######################################################

docker/Dockerfile.amd

+1
Original file line numberDiff line numberDiff line change
@@ -35,6 +35,7 @@ RUN rm /etc/apt/sources.list.d/* \
3535
gfortran \
3636
flex \
3737
hwloc \
38+
gdb \
3839
&& apt-get clean
3940

4041
# Requires cmake > 3.22

docker/Dockerfile.cpu

+4-3
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@ RUN apt-get update && \
1717

1818
# Install python
1919
RUN apt-get update && \
20-
apt-get install -y dh-autoreconf python3-venv python3-dev python3-pip
20+
apt-get install -y dh-autoreconf python3-venv python3-dev python3-pip gdb
2121

2222
# Install for basic base not containing it
2323
RUN apt-get install -y vim wget git flex libnuma-dev tmux \
@@ -27,14 +27,14 @@ RUN apt-get install -y vim wget git flex libnuma-dev tmux \
2727

2828
# Install compilersx
2929
RUN if [ -n "$gcc" ]; then \
30-
apt-get install libgmp-dev libmpc-dev libmpfr-dev libisl-dev -y && \
30+
apt-get install libgmp-dev libmpc-dev libmpfr-dev libisl-dev binutils texinfo -y && \
3131
cd /tmp && mkdir gcc && \
3232
git clone git://gcc.gnu.org/git/gcc.git gcc && \
3333
cd gcc && git checkout releases/gcc-${gcc} && \
3434
mkdir build && cd build && \
3535
../configure --prefix=/opt/gcc-${gcc} --enable-languages=c,c++,fortran --disable-multilib \
3636
--enable-shared --enable-lto --enable-threads=posix --enable-libstdcxx-time=yes --enable-libgomp && \
37-
make -j $(nproc) && make install && \
37+
make -j $(nproc) && make install-strip && \
3838
# Update alternatives
3939
update-alternatives --install /usr/bin/gcc gcc /opt/gcc-${gcc}/bin/gcc 60 && \
4040
update-alternatives --install /usr/bin/g++ g++ /opt/gcc-${gcc}/bin/g++ 60 && \
@@ -59,6 +59,7 @@ RUN cd /tmp && mkdir openmpi && \
5959
cd openmpi && ./autogen.pl && \
6060
mkdir build && cd build && \
6161
../configure --prefix=/opt/openmpi/ \
62+
--disable-mpi-fortran \
6263
--enable-mca-no-build=btl-uct --enable-mpi1-compatibility && \
6364
make -j ${nproc} && \
6465
make install && \

docker/Dockerfile.intel

+8-1
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@ ENV DEBIAN_FRONTEND noninteractive
1010

1111
# Install python
1212
RUN apt-get update && \
13-
apt-get install -y dh-autoreconf python3-venv python3-dev python3-pip
13+
apt-get install -y dh-autoreconf python3-venv python3-dev python3-pip gdb
1414

1515
# Install for basic base not containing it
1616
RUN apt-get install -y vim wget git flex libnuma-dev tmux \
@@ -56,9 +56,16 @@ RUN apt-get update -y && apt-get dist-upgrade -y && \
5656
libegl-mesa0 libegl1-mesa libegl1-mesa-dev libgbm1 libgl1-mesa-dev libgl1-mesa-dri \
5757
libglapi-mesa libgles2-mesa-dev libglx-mesa0 libigdgmm12 libxatracker2 mesa-va-drivers \
5858
mesa-vdpau-drivers mesa-vulkan-drivers va-driver-all vainfo hwinfo clinfo \
59+
# XPU-smi requirements
60+
intel-gsc libmfx-tools intel-metrics-discovery intel-metrics-library \
5961
# Development packages
6062
libigc-dev intel-igc-cm libigdfcl-dev libigfxcmrt-dev level-zero-dev
6163

64+
# Install xpu-smi for device detection
65+
RUN wget https://github.com/intel/xpumanager/releases/download/V1.2.39/xpu-smi_1.2.39_20240906.085820.11f3c29a+deb10u1_amd64.deb && \
66+
apt install -y ./xpu-smi_1.2.39_20240906.085820.11f3c29a+deb10u1_amd64.deb && \
67+
rm xpu-smi_1.2.39_20240906.085820.11f3c29a+deb10u1_amd64.deb
68+
6269
ENV MPI4PY_FLAGS='. /opt/intel/oneapi/setvars.sh intel64 && '
6370
ENV MPI4PY_RC_RECV_MPROBE=0
6471

docker/Dockerfile.nvidia

+1-1
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@ ENV DEBIAN_FRONTEND noninteractive
1616
# Install python
1717
RUN apt-get update && \
1818
apt-get install -y -q gpg apt-utils curl wget vim libnuma-dev tmux numactl \
19-
dh-autoreconf python3-venv python3-dev python3-pip
19+
dh-autoreconf python3-venv python3-dev python3-pip gdb
2020

2121
# nodesource: nvdashboard requires nodejs>=10
2222
RUN curl https://developer.download.nvidia.com/hpc-sdk/ubuntu/DEB-GPG-KEY-NVIDIA-HPC-SDK | gpg --yes --dearmor -o /usr/share/keyrings/nvidia-hpcsdk-archive-keyring.gpg

0 commit comments

Comments
 (0)