Skip to content
Snippets Groups Projects
Commit a95bd7bf 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 195f37ff
No related branches found
No related tags found
No related merge requests found
...@@ -250,7 +250,11 @@ ENV CLICOLOR_FORCE=1 \ ...@@ -250,7 +250,11 @@ ENV CLICOLOR_FORCE=1 \
## Install additional packages. Remove the auto-cleanup functionality ## Install additional packages. Remove the auto-cleanup functionality
## for docker, as we're using the new buildkit cache instead. ## 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 ## We install gcc-10 and g++-10 as they are not installed by default in
<<<<<<< HEAD
## Ubuntu 20.04. ## Ubuntu 20.04.
=======
## Ubuntu 20.04. Install CUDA runtime libraries.
>>>>>>> Add CUDA libraries
## TODO: libyaml-cpp-dev is a dependency for afterburner. We can probably remove ## TODO: libyaml-cpp-dev is a dependency for afterburner. We can probably remove
## this once afterburner is added to spack ## this once afterburner is added to spack
RUN --mount=type=cache,target=/var/cache/apt \ RUN --mount=type=cache,target=/var/cache/apt \
...@@ -322,12 +326,17 @@ RUN --mount=type=cache,target=/var/cache/apt \ ...@@ -322,12 +326,17 @@ RUN --mount=type=cache,target=/var/cache/apt \
cuda-cudart-11-7 \ cuda-cudart-11-7 \
cuda-compat-11-7 \ cuda-compat-11-7 \
&& apt-get -yqq autoremove \ && apt-get -yqq autoremove \
<<<<<<< HEAD
&& ln -s cuda-11.7 /usr/local/cuda \ && ln -s cuda-11.7 /usr/local/cuda \
&& rm -rf /var/lib/apt/lists/* && rm -rf /var/lib/apt/lists/*
ENV PATH /usr/local/nvidia/bin:/usr/local/cuda/bin:${PATH} ENV PATH /usr/local/nvidia/bin:/usr/local/cuda/bin:${PATH}
ENV LD_LIBRARY_PATH /usr/local/nvidia/lib:/usr/local/nvidia/lib64 ENV LD_LIBRARY_PATH /usr/local/nvidia/lib:/usr/local/nvidia/lib64
=======
&& rm -rf /var/lib/apt/lists/*
>>>>>>> Add CUDA libraries
## copy over everything we need from staging in a single layer ## copy over everything we need from staging in a single layer
RUN --mount=from=staging,target=/staging \ RUN --mount=from=staging,target=/staging \
rm -rf /usr/local \ 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