Skip to content

Commit

Permalink
Add Ubuntu 22.04 image with CUDA
Browse files Browse the repository at this point in the history
This commit adds a new Ubuntu 22.04 image with all the ACTS
dependencies, and an installation of CUDA 12.4.
  • Loading branch information
stephenswat committed Jun 10, 2024
1 parent 7680d98 commit 9a3fb81
Showing 1 changed file with 33 additions and 0 deletions.
33 changes: 33 additions & 0 deletions ubuntu2204_cuda/Dockerfile
Original file line number Diff line number Diff line change
@@ -0,0 +1,33 @@
# Docker machinery, part of the ACTS project
#
# (c) 2024 CERN for the benefit of the ACTS project
#
# Mozilla Public License Version 2.0

# Start from the (at the time of writing) latest Acts Ubuntu 22.04 image.
FROM ghcr.io/acts-project/ubuntu2204:v47

# Some description for the image.
LABEL description="Ubuntu 22.04 with Acts dependencies and CUDA"
LABEL maintainer="Stephen Nicholas Swatman <[email protected]"

# Add the Ubuntu 20.04 CUDA repository.
RUN curl -SL https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.1-1_all.deb \
-o cuda-keyring.deb && \
dpkg -i cuda-keyring.deb && \
rm cuda-keyring.deb

# Install CUDA.
ARG CUDA_VERSION=12-4
RUN apt-get update && \
apt-get install -y cuda-compat-${CUDA_VERSION} cuda-cudart-${CUDA_VERSION} \
cuda-libraries-${CUDA_VERSION} \
cuda-command-line-tools-${CUDA_VERSION} \
cuda-minimal-build-${CUDA_VERSION} && \
apt-get clean -y

# Set up the CUDA environment.
ENV NVIDIA_VISIBLE_DEVICES=all
ENV NVIDIA_DRIVER_CAPABILITIES=compute,utility
ENV PATH=/usr/local/cuda/bin:${PATH}
ENV LD_LIBRARY_PATH=/usr/local/cuda/lib64:${LD_LIBRARY_PATH}

0 comments on commit 9a3fb81

Please sign in to comment.