Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

📌 Update CUDA to 12.6.1 #19

Merged
merged 1 commit into from
Sep 24, 2024
Merged

📌 Update CUDA to 12.6.1 #19

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
16 changes: 8 additions & 8 deletions Dockerfile
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
#checkov:skip=CKV_DOCKER_2: HEALTHCHECK not required - Health checks are implemented downstream of this image

FROM public.ecr.aws/ubuntu/ubuntu@sha256:65ccda647ad998c36c5b0365e308ec0b1bc770ace445d5e954d55ac8c19a9c27
FROM public.ecr.aws/ubuntu/ubuntu@sha256:5b2fc4131b3c134a019c3ea815811de70e6ad9ee1626f59bf302558a95b436e5

LABEL org.opencontainers.image.vendor="Ministry of Justice" \
org.opencontainers.image.authors="Analytical Platform (analytical-platform@digital.justice.gov.uk)" \
Expand All @@ -15,10 +15,10 @@ ENV CONTAINER_USER="analyticalplatform" \
ANALYTICAL_PLATFORM_DIRECTORY="/opt/analytical-platform" \
DEBIAN_FRONTEND="noninteractive" \
PIP_BREAK_SYSTEM_PACKAGES="1" \
CUDA_VERSION="12.5.1" \
CUDA_VERSION="12.6.1" \
NVIDIA_DISABLE_REQUIRE="true" \
NVIDIA_CUDA_CUDART_VERSION="12.5.82-1" \
NVIDIA_CUDA_COMPAT_VERSION="555.42.06-1" \
NVIDIA_CUDA_CUDART_VERSION="12.6.68-1" \
NVIDIA_CUDA_COMPAT_VERSION="560.35.03-1" \
NVIDIA_VISIBLE_DEVICES="all" \
NVIDIA_DRIVER_CAPABILITIES="compute,utility" \
LD_LIBRARY_PATH="/usr/local/nvidia/lib:/usr/local/nvidia/lib64" \
Expand Down Expand Up @@ -49,10 +49,10 @@ apt-get update --yes
apt-get install --yes \
"apt-transport-https=2.7.14build2" \
"ca-certificates=20240203" \
"curl=8.5.0-2ubuntu10.3" \
"curl=8.5.0-2ubuntu10.4" \
"git=1:2.43.0-1ubuntu7.1" \
"jq=1.7.1-3build1" \
"python3.12=3.12.3-1ubuntu0.1" \
"python3.12=3.12.3-1ubuntu0.2" \
"python3-pip=24.0+dfsg-1ubuntu1" \
"unzip=6.0-28ubuntu4"

Expand All @@ -78,8 +78,8 @@ echo "deb [signed-by=/etc/apt/keyrings/nvidia.gpg] https://developer.download.nv
apt-get update --yes

apt-get install --yes \
"cuda-cudart-12-5=${NVIDIA_CUDA_CUDART_VERSION}" \
"cuda-compat-12-5=${NVIDIA_CUDA_COMPAT_VERSION}"
"cuda-cudart-12-6=${NVIDIA_CUDA_CUDART_VERSION}" \
"cuda-compat-12-6=${NVIDIA_CUDA_COMPAT_VERSION}"

echo "/usr/local/nvidia/lib" >> /etc/ld.so.conf.d/nvidia.conf
echo "/usr/local/nvidia/lib64" >> /etc/ld.so.conf.d/nvidia.conf
Expand Down