From 5934e1ea41acf1fb773d0a76bfbea34e7dc17307 Mon Sep 17 00:00:00 2001
From: Ujwal Kundur <ujwal.kundur@gmail.com>
Date: Thu, 18 Aug 2022 11:59:00 +0530
Subject: [PATCH] Add CUDA libraries
Install CUDA Native Runtime Libraries - cudart &&
CUDA Compatibility packages for Driver support - cuda-compat
All CUDA libraries are based on version 11.7
---
containers/oneapi/onedev.Dockerfile | 14 +++++++++++++-
1 file changed, 13 insertions(+), 1 deletion(-)
diff --git a/containers/oneapi/onedev.Dockerfile b/containers/oneapi/onedev.Dockerfile
index d24a7a8d2..382de16ef 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 \
--
GitLab