Skip to content
Snippets Groups Projects
Commit 4d3ec737 authored by Ujwal Kundur's avatar Ujwal Kundur
Browse files

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
parent c3607c85
No related branches found
No related tags found
No related merge requests found
......@@ -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 \
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment