Skip to content

Commit

Permalink
Add new Ubuntu 24.04 CUDA, ROCm, OneAPI machines
Browse files Browse the repository at this point in the history
This commit adds three new images based on Ubuntu 24.04, with OneAPI
and, optionally, the AMD ROCm and NVIDIA CUDA plugins. Also adds a new
Ubuntu 22.04 image with the dependencies built using C++20.
  • Loading branch information
stephenswat committed Jul 22, 2024
1 parent 6e9b223 commit 7c7016b
Show file tree
Hide file tree
Showing 11 changed files with 375 additions and 0 deletions.
4 changes: 4 additions & 0 deletions .github/workflows/build-images.yml
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,7 @@ on:
push:
branches:
- main
- "*"
tags:
- 'v*'

Expand All @@ -26,6 +27,9 @@ jobs:
- ubuntu2204_rocm_clang
- ubuntu2404
- ubuntu2404_cuda
- ubuntu2404_cuda_oneapi
- ubuntu2404_rocm_oneapi
- ubuntu2404_oneapi
- alma9-base
steps:
- uses: actions/checkout@v4
Expand Down
182 changes: 182 additions & 0 deletions ubuntu2204_rocm_clang_cxx20/Dockerfile
Original file line number Diff line number Diff line change
@@ -0,0 +1,182 @@
# Docker machinery, part of the ACTS project
#
# (c) 2024 CERN for the benefit of the ACTS project
#
# Mozilla Public License Version 2.0

FROM rocm/dev-ubuntu-22.04:6.1

LABEL description="Ubuntu 22.04 with Acts dependencies and ROCm using clang, built using C++20"
LABEL maintainer="Paul Gessinger <[email protected]>"
# increase whenever any of the RUN commands change
LABEL version="1"

# DEBIAN_FRONTEND ensures non-blocking operation (tzdata is a problem)
ENV DEBIAN_FRONTEND noninteractive

# install dependencies from the package manager.
#
# see also https://root.cern.ch/build-prerequisites
RUN apt-get update -y \
&& apt-get install -y \
build-essential \
clang \
cmake \
curl \
git \
freeglut3-dev \
libboost-dev \
libboost-filesystem-dev \
libboost-program-options-dev \
libboost-test-dev \
libeigen3-dev \
libexpat-dev \
libftgl-dev \
libgl2ps-dev \
libglew-dev \
libgsl-dev \
liblz4-dev \
liblzma-dev \
libpcre3-dev \
libtbb-dev \
libx11-dev \
libxext-dev \
libxft-dev \
libxpm-dev \
libxerces-c-dev \
libxxhash-dev \
libzstd-dev \
ninja-build \
python3 \
python3-dev \
python3-pip \
rsync \
zlib1g-dev \
ccache \
&& apt-get remove -y gcc g++ \
&& apt-get clean -y

# manual builds for hep-specific packages
ENV GET curl --location --silent --create-dirs
ENV UNPACK_TO_SRC tar -xz --strip-components=1 --directory src
ENV PREFIX /usr/local

# use clang by default
ENV CC clang
ENV CXX clang++

# Geant4
RUN mkdir src \
&& ${GET} https://gitlab.cern.ch/geant4/geant4/-/archive/v11.1.1/geant4-v11.1.1.tar.gz \
| ${UNPACK_TO_SRC} \
&& cmake -B build -S src -GNinja \
-DCMAKE_BUILD_TYPE=Release \
-DCMAKE_INSTALL_PREFIX=${PREFIX} \
-DCMAKE_CXX_STANDARD=20 \
-DGEANT4_BUILD_TLS_MODEL=global-dynamic \
-DGEANT4_INSTALL_DATA=OFF \
-DGEANT4_USE_GDML=ON \
-DGEANT4_USE_SYSTEM_EXPAT=ON \
-DGEANT4_USE_SYSTEM_ZLIB=ON \
&& cmake --build build -- install \
&& rm -rf build src

# HepMC3
RUN mkdir src \
&& ${GET} https://gitlab.cern.ch/hepmc/HepMC3/-/archive/3.2.5/HepMC3-3.2.5.tar.gz \
| ${UNPACK_TO_SRC} \
&& cmake -B build -S src -GNinja \
-DCMAKE_BUILD_TYPE=Release \
-DCMAKE_INSTALL_PREFIX=${PREFIX} \
-DHEPMC3_BUILD_STATIC_LIBS=OFF \
-DHEPMC3_ENABLE_PYTHON=OFF \
-DHEPMC3_ENABLE_ROOTIO=OFF \
-DHEPMC3_ENABLE_SEARCH=OFF \
&& cmake --build build -- install \
&& rm -rf build src

# Pythia8
# requires rsync; installation uses `rsync` instead of `install`
RUN mkdir src \
&& ${GET} https://pythia.org/download/pythia83/pythia8309.tgz\
| ${UNPACK_TO_SRC} \
&& cd src \
&& ./configure --enable-shared --prefix=${PREFIX} \
&& make -j$(nproc) install \
&& cd .. \
&& rm -rf src

# nlohmann's JSON
RUN mkdir src \
&& ${GET} https://github.com/nlohmann/json/archive/refs/tags/v3.11.2.tar.gz \
| ${UNPACK_TO_SRC} \
&& cmake -B build -S src -GNinja \
-DJSON_BuildTests=OFF \
&& cmake --build build -- install \
&& rm -rf build src

# ROOT
RUN mkdir src \
&& ${GET} https://root.cern/download/root_v6.28.02.source.tar.gz \
| ${UNPACK_TO_SRC} \
&& cmake -B build -S src -GNinja \
-DCMAKE_BUILD_TYPE=Release \
-DCMAKE_CXX_STANDARD=20 \
-DCMAKE_INSTALL_PREFIX=${PREFIX} \
-Dfail-on-missing=ON \
-Dgminimal=ON \
-Dgdml=ON \
-Dopengl=ON \
-Dpyroot=ON \
&& cmake --build build -- install \
&& rm -rf build src

# environment variables needed to find ROOT libraries
ENV LD_LIBRARY_PATH /usr/local/lib
ENV PYTHON_PATH /usr/local/lib

# podio
RUN mkdir src \
&& ${GET} https://github.com/AIDASoft/podio/archive/refs/tags/v00-16-03.tar.gz \
| ${UNPACK_TO_SRC} \
&& cmake -B build -S src -GNinja \
-DCMAKE_BUILD_TYPE=Release \
-DCMAKE_INSTALL_PREFIX=${PREFIX} \
-DBUILD_TESTING=OFF \
-USE_EXTERNAL_CATCH2=OFF \
&& cmake --build build -- install \
&& rm -rf build src

# EDM4hep
RUN pip3 install jinja2 pyyaml \
&& mkdir src \
&& ${GET} https://github.com/key4hep/EDM4hep/archive/refs/tags/v00-07-02.tar.gz \
| ${UNPACK_TO_SRC} \
&& cmake -B build -S src -GNinja \
-DCMAKE_BUILD_TYPE=Release \
-DCMAKE_INSTALL_PREFIX=${PREFIX} \
-DBUILD_TESTING=OFF \
-DUSE_EXTERNAL_CATCH2=OFF \
&& cmake --build build -- install \
&& rm -rf build src

# DD4hep
# requires Geant4 and ROOT and must come last
RUN mkdir src \
&& ${GET} https://github.com/AIDASoft/DD4hep/archive/v01-25-01.tar.gz \
| ${UNPACK_TO_SRC} \
&& cmake -B build -S src -GNinja \
-DCMAKE_BUILD_TYPE=Release \
-DCMAKE_CXX_STANDARD=20 \
-DCMAKE_INSTALL_PREFIX=${PREFIX} \
-DCMAKE_PREFIX_PATH=${PREFIX} \
-DBUILD_TESTING=OFF \
-DDD4HEP_BUILD_PACKAGES="DDG4 DDDetectors DDRec UtilityApps" \
-DDD4HEP_IGNORE_GEANT4_TLS=ON \
-DDD4HEP_USE_GEANT4=ON \
-DDD4HEP_USE_XERCESC=ON \
-DDD4HEP_USE_EDM4HEP=ON \
&& cmake --build build -- install \
&& rm -rf build src

COPY download_geant4_data.sh /usr/local/bin
26 changes: 26 additions & 0 deletions ubuntu2204_rocm_clang_cxx20/download_geant4_data.sh
Original file line number Diff line number Diff line change
@@ -0,0 +1,26 @@
#!/bin/bash
set -e

data_dir=/usr/local/share/Geant4-10.6.1/data

mkdir $data_dir
cd $data_dir


function dl {
url=$1
echo "Downloading $url"
curl --location $url | tar -xz
}

dl https://geant4-data.web.cern.ch/datasets/G4NDL.4.6.tar.gz
dl https://geant4-data.web.cern.ch/datasets/G4EMLOW.7.9.1.tar.gz
dl https://geant4-data.web.cern.ch/datasets/G4PhotonEvaporation.5.5.tar.gz
dl https://geant4-data.web.cern.ch/datasets/G4RadioactiveDecay.5.4.tar.gz
dl https://geant4-data.web.cern.ch/datasets/G4PARTICLEXS.2.1.tar.gz
dl https://geant4-data.web.cern.ch/datasets/G4PII.1.3.tar.gz
dl https://geant4-data.web.cern.ch/datasets/G4RealSurface.2.1.1.tar.gz
dl https://geant4-data.web.cern.ch/datasets/G4SAIDDATA.2.0.tar.gz
dl https://geant4-data.web.cern.ch/datasets/G4ABLA.3.1.tar.gz
dl https://geant4-data.web.cern.ch/datasets/G4INCL.1.0.tar.gz
dl https://geant4-data.web.cern.ch/datasets/G4ENSDFSTATE.2.2.tar.gz
65 changes: 65 additions & 0 deletions ubuntu2404_cuda_oneapi/Dockerfile
Original file line number Diff line number Diff line change
@@ -0,0 +1,65 @@
# Docker machinery, part of the ACTS project
#
# (c) 2024 CERN for the benefit of the ACTS project
#
# Mozilla Public License Version 2.0

# Start from the (at the time of writing) latest Acts Ubuntu 24.04 image.
FROM ghcr.io/acts-project/ubuntu2404:53

# Some description for the image.
LABEL description="Ubuntu 24.04 with Acts dependencies and CUDA + oneAPI"
LABEL maintainer="Stephen Nicholas Swatman <[email protected]>"

# Add the Ubuntu 24.04 CUDA repository.
RUN curl -SL https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2404/x86_64/cuda-keyring_1.1-1_all.deb \
-o cuda-keyring.deb && \
dpkg -i cuda-keyring.deb && \
rm cuda-keyring.deb

# Install CUDA.
ARG CUDA_VERSION=12-5
RUN apt-get update && \
apt-get install -y cuda-compat-${CUDA_VERSION} cuda-cudart-${CUDA_VERSION} \
cuda-libraries-dev-${CUDA_VERSION} \
cuda-command-line-tools-${CUDA_VERSION} \
cuda-minimal-build-${CUDA_VERSION} && \
apt-get clean -y

# Set up the CUDA environment.
ENV NVIDIA_VISIBLE_DEVICES=all
ENV NVIDIA_DRIVER_CAPABILITIES=compute,utility
ENV PATH=/usr/local/cuda/bin:${PATH}
ENV LD_LIBRARY_PATH=/usr/local/cuda/lib64:${LD_LIBRARY_PATH}
ENV CUDAHOSTCXX="clang++"
ENV CUDAFLAGS="-allow-unsupported-compiler"

# Set up the Intel package signing key.
RUN mkdir --parents --mode=0755 /etc/apt/keyrings && \
curl -SL https://apt.repos.intel.com/intel-gpg-keys/GPG-PUB-KEY-INTEL-SW-PRODUCTS.PUB | \
gpg --dearmor > /usr/share/keyrings/oneapi-archive-keyring.gpg

# Set up the oneAPI repository.
COPY oneapi.list /etc/apt/sources.list.d/

# Install oneAPI.
ARG ONEAPI_VERSION=2024.2
RUN apt-get update && \
apt-get install -y intel-oneapi-compiler-dpcpp-cpp-${ONEAPI_VERSION} && \
apt-get clean -y

# Install the CodePlay NVIDIA plugin on top of oneAPI.
ARG CODEPLAY_PLUGIN_VERSION=2024.2
RUN curl -SL \
"https://developer.codeplay.com/api/v1/products/download?product=oneapi&variant=nvidia&version=${CODEPLAY_PLUGIN_VERSION}" \
-o plugin.sh && \
sh plugin.sh --install-dir /opt/intel/oneapi --yes && \
rm plugin.sh

# Set up the oneAPI environment. Note that one *MUST* source the
# /opt/intel/oneapi/setvars.sh script to make the following work. Which has to
# be done in the respective CI scripts.
ENV CC="icx"
ENV CXX="icpx"
ENV SYCLCXX="icpx"
ENV SYCLFLAGS="-fsycl -fsycl-targets=nvidia_gpu_sm_75 -Wno-unknown-cuda-version"
1 change: 1 addition & 0 deletions ubuntu2404_cuda_oneapi/oneapi.list
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
deb [signed-by=/usr/share/keyrings/oneapi-archive-keyring.gpg] https://apt.repos.intel.com/oneapi all main
34 changes: 34 additions & 0 deletions ubuntu2404_oneapi/Dockerfile
Original file line number Diff line number Diff line change
@@ -0,0 +1,34 @@
# Docker machinery, part of the ACTS project
#
# (c) 2024 CERN for the benefit of the ACTS project
#
# Mozilla Public License Version 2.0

# Start from the (at the time of writing) latest Acts Ubuntu 24.04 image.
FROM ghcr.io/acts-project/ubuntu2404:53

# Some description for the image.
LABEL description="Ubuntu 24.04 with Acts dependencies and oneAPI"
LABEL maintainer="Stephen Nicholas Swatman <[email protected]>"

# Set up the Intel package signing key.
RUN mkdir --parents --mode=0755 /etc/apt/keyrings && \
curl -SL https://apt.repos.intel.com/intel-gpg-keys/GPG-PUB-KEY-INTEL-SW-PRODUCTS.PUB | \
gpg --dearmor > /usr/share/keyrings/oneapi-archive-keyring.gpg

# Set up the oneAPI repository.
COPY oneapi.list /etc/apt/sources.list.d/

# Install oneAPI.
ARG ONEAPI_VERSION=2024.2
RUN apt-get update && \
apt-get install -y intel-oneapi-compiler-dpcpp-cpp-${ONEAPI_VERSION} && \
apt-get clean -y

# Set up the environment. Note that one *MUST* source the
# /opt/intel/oneapi/setvars.sh script to make the following work. Which has to
# be done in the respective CI scripts.
ENV CC="icx"
ENV CXX="icpx"
ENV SYCLCXX="icpx"
ENV SYCLFLAGS="-fsycl -fsycl-targets=spir64,spir64_x86_64"
1 change: 1 addition & 0 deletions ubuntu2404_oneapi/oneapi.list
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
deb [signed-by=/usr/share/keyrings/oneapi-archive-keyring.gpg] https://apt.repos.intel.com/oneapi all main
57 changes: 57 additions & 0 deletions ubuntu2404_rocm_oneapi/Dockerfile
Original file line number Diff line number Diff line change
@@ -0,0 +1,57 @@
# Docker machinery, part of the ACTS project
#
# (c) 2024 CERN for the benefit of the ACTS project
#
# Mozilla Public License Version 2.0

# Start from the (at the time of writing) latest Acts Ubuntu 24.04 image.
FROM ghcr.io/acts-project/ubuntu2404:53

# Some description for the image.
LABEL description="Ubuntu 24.04 with Acts dependencies and ROCm/HIP + oneAPI"
LABEL maintainer="Stephen Nicholas Swatman <[email protected]>"

# Set up the ROCm package signing key.
RUN mkdir --parents --mode=0755 /etc/apt/keyrings && \
curl -SL https://repo.radeon.com/rocm/rocm.gpg.key | \
gpg --dearmor > /etc/apt/keyrings/rocm.gpg

# Set up the ROCm repository.
COPY rocm.list /etc/apt/sources.list.d/
COPY rocm-pin-600 /etc/apt/preferences.d/

# Install ROCm/HIP.
ARG ROCM_VERSION=6.1.0
RUN apt-get update && \
apt-get install -y rocm-hip-runtime-dev${ROCM_VERSION} && \
apt-get clean -y
ENV HIP_PLATFORM=amd

# Set up the Intel package signing key.
RUN curl -SL https://apt.repos.intel.com/intel-gpg-keys/GPG-PUB-KEY-INTEL-SW-PRODUCTS.PUB | \
gpg --dearmor > /usr/share/keyrings/oneapi-archive-keyring.gpg

# Set up the oneAPI repository.
COPY oneapi.list /etc/apt/sources.list.d/

# Install oneAPI.
ARG ONEAPI_VERSION=2024.2
RUN apt-get update && \
apt-get install -y intel-oneapi-compiler-dpcpp-cpp-${ONEAPI_VERSION} && \
apt-get clean -y

# Install the CodePlay AMD plugin on top of oneAPI.
ARG CODEPLAY_PLUGIN_VERSION=2024.2.0
RUN curl -SL \
"https://developer.codeplay.com/api/v1/products/download?product=oneapi&variant=amd&version=${CODEPLAY_PLUGIN_VERSION}" \
-o plugin.sh && \
sh plugin.sh --install-dir /opt/intel/oneapi --yes && \
rm plugin.sh

# Set up the oneAPI environment. Note that one *MUST* source the
# /opt/intel/oneapi/setvars.sh script to make the following work. Which has to
# be done in the respective CI scripts.
ENV CC="icx"
ENV CXX="icpx"
ENV SYCLCXX="icpx"
ENV SYCLFLAGS="-fsycl -fsycl-targets=amd_gpu_gfx1031"
1 change: 1 addition & 0 deletions ubuntu2404_rocm_oneapi/oneapi.list
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
deb [signed-by=/usr/share/keyrings/oneapi-archive-keyring.gpg] https://apt.repos.intel.com/oneapi all main
3 changes: 3 additions & 0 deletions ubuntu2404_rocm_oneapi/rocm-pin-600
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
Package: *
Pin: release o=repo.radeon.com
Pin-Priority: 600
1 change: 1 addition & 0 deletions ubuntu2404_rocm_oneapi/rocm.list
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
deb [arch=amd64 signed-by=/etc/apt/keyrings/rocm.gpg] https://repo.radeon.com/rocm/apt/6.1 focal main

0 comments on commit 7c7016b

Please sign in to comment.