Use nvhpc 24.5 and pull on host machine

C2SM · May 30, 2024 · 2627ee8 · 2627ee8
1 parent c4dd174
commit 2627ee8
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 7 deletions.
diff --git a/ci/base.yml b/ci/base.yml
@@ -44,3 +44,4 @@ variables:
     CSCS_NEEDED_DATA: icon4py
     TEST_DATA_PATH: "/project/d121/icon4py/ci/testdata"
     ICON_GRID_LOC: "/project/d121/icon4py/ci/testdata/grids/mch_ch_r04b09_dsl"
+    PULL_ON_HOST_MACHINE: 'YES'
diff --git a/ci/docker/Dockerfile.build b/ci/docker/Dockerfile.build
@@ -31,9 +31,9 @@ RUN apt-get update -qq && apt-get install -qq -y --no-install-recommends \
     rm -rf /var/lib/apt/lists/*
 
 # Install NVIDIA HPC SDK for nvfortran
-ARG HPC_SDK_VERSION=22.11
-ARG HPC_SDK_NAME=nvhpc_2022_2211_Linux_x86_64_cuda_11.8
-ARG HPC_SDK_URL=https://developer.download.nvidia.com/hpc-sdk/22.11/${HPC_SDK_NAME}.tar.gz
+ARG HPC_SDK_VERSION=24.5
+ARG HPC_SDK_NAME=nvhpc_2024_245_Linux_x86_64_cuda_multi
+ARG HPC_SDK_URL=https://developer.download.nvidia.com/hpc-sdk/24.5/${HPC_SDK_NAME}.tar.gz
 
 RUN wget -q ${HPC_SDK_URL} -O /tmp/nvhpc.tar.gz && \
     mkdir -p /opt/nvidia && \
@@ -78,7 +78,4 @@ ENV PATH="/root/.pyenv/shims:${PATH}"
 
 # Install Python packages
 COPY . /icon4py
-RUN pip install --upgrade pip setuptools wheel tox clang-format
-
-# Install cupy after NVIDIA HPC SDK is installed
-RUN pip install cupy-cuda11x
+RUN pip install --upgrade pip setuptools wheel tox clang-format cupy-cuda11x