diff --git a/containers/oneapi/onedev.Dockerfile b/containers/oneapi/onedev.Dockerfile index d24a7a8d2122efa8f093258b6abcd716dff49b23..382de16ef7d091e407092c9a14963055423e460f 100644 --- a/containers/oneapi/onedev.Dockerfile +++ b/containers/oneapi/onedev.Dockerfile @@ -250,7 +250,7 @@ ENV CLICOLOR_FORCE=1 \ ## Install additional packages. Remove the auto-cleanup functionality ## for docker, as we're using the new buildkit cache instead. ## We install gcc-10 and g++-10 as they are not installed by default in -## Ubuntu 20.04 +## Ubuntu 20.04. Install CUDA runtime libraries. ## TODO: libyaml-cpp-dev is a dependency for afterburner. We can probably remove ## this once afterburner is added to spack RUN --mount=type=cache,target=/var/cache/apt \ @@ -312,6 +312,18 @@ RUN --mount=type=cache,target=/var/cache/apt \ && apt-get -yqq autoremove \ && rm -rf /var/lib/apt/lists/* +## Install CUDA Libraries +RUN --mount=type=cache,target=/var/cache/apt \ + wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-keyring_1.0-1_all.deb \ + && dpkg -i cuda-keyring_1.0-1_all.deb \ + && apt-get -yqq update \ + && apt-get -yqq upgrade \ + && apt-get -yqq install --no-install-recommends \ + cuda-cudart-11-7 \ + cuda-compat-11-7 \ + && apt-get -yqq autoremove \ + && rm -rf /var/lib/apt/lists/* + ## copy over everything we need from staging in a single layer RUN --mount=from=staging,target=/staging \ rm -rf /usr/local \