Skip to content

Commit

Permalink
fix: Zk-environment proper Nvidia driver repos (#3471)
Browse files Browse the repository at this point in the history
## What ❔

Change of deprecated nvidia repos to newer ones, and changing from
ubuntu 20 repos to ubuntu 22

## Why ❔



## Checklist

<!-- Check your PR fulfills the following items. -->
<!-- For draft PRs check the boxes as you complete them. -->

- [x] PR title corresponds to the body of PR (we generate changelog
entries from PRs).
- [x] Tests for the changes have been added / updated.
- [x] Documentation comments have been added / updated.
- [x] Code has been formatted via `zkstack dev fmt` and `zkstack dev
lint`.
  • Loading branch information
artmakh authored Jan 14, 2025
1 parent 84e3e31 commit 006691f
Show file tree
Hide file tree
Showing 2 changed files with 10 additions and 5 deletions.
9 changes: 7 additions & 2 deletions docker/zk-environment/22.04_amd64_cuda_11_8.Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -123,8 +123,13 @@ ENV NVIDIA_REQUIRE_CUDA "cuda>=11.8 brand=tesla,driver>=450,driver<451 brand=tes
ENV NV_CUDA_CUDART_VERSION 11.8.89-1
ENV NV_CUDA_COMPAT_PACKAGE cuda-compat-11-8

RUN wget -c -O - https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/${NVARCH}/3bf863cc.pub | apt-key add - && \
echo "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/${NVARCH} /" > /etc/apt/sources.list.d/cuda.list
# curl purging is removed, it's required in next steps
RUN apt-get update && apt-get install -y --no-install-recommends \
gnupg2 curl ca-certificates && \
curl -fsSLO https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/${NVARCH}/cuda-keyring_1.0-1_all.deb && \
dpkg -i cuda-keyring_1.0-1_all.deb && \
apt-get purge --autoremove -y curl \
&& rm -rf /var/lib/apt/lists/*

ENV CUDA_VERSION 11.8.0

Expand Down
6 changes: 3 additions & 3 deletions docker/zk-environment/22.04_amd64_cuda_12.Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -127,9 +127,9 @@ ENV NV_CUDA_COMPAT_PACKAGE cuda-compat-12-2
# curl purging is removed, it's required in next steps
RUN apt-get update && apt-get install -y --no-install-recommends \
gnupg2 curl ca-certificates && \
wget -c -O - https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/${NVARCH}/3bf863cc.pub | apt-key add - && \
echo "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/${NVARCH} /" > /etc/apt/sources.list.d/cuda.list && \
rm -rf /var/lib/apt/lists/*
curl -fsSLO https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/${NVARCH}/cuda-keyring_1.0-1_all.deb && \
dpkg -i cuda-keyring_1.0-1_all.deb && \
&& rm -rf /var/lib/apt/lists/*

ENV CUDA_VERSION 12.2.2

Expand Down

0 comments on commit 006691f

Please sign in to comment.