Skip to content

Commit

Permalink
fix(container): add NVIDIA GPG key
Browse files Browse the repository at this point in the history
for usage with APT repositories in stage 2 of the build. This makes
it easier for users to re-use the container image as-is without
figuring out how to inject the new GPG key from NVIDIA.

Resolves k4yt3x#1013

Signed-off-by: Luke Short <[email protected]>
  • Loading branch information
LukeShortCloud committed Nov 5, 2023
1 parent 26cf8da commit 8b92bd5
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -9,6 +9,8 @@ ENV DEBIAN_FRONTEND=noninteractive

COPY . /video2x
WORKDIR /video2x
# Manually add the new NVIDIA GPG keys since the docker.io/nvidia/vulkan container is old.
# https://github.com/NVIDIA/nvidia-docker/issues/1632
RUN gpg --keyserver=keyserver.ubuntu.com --receive-keys A4B469963BF863CC \
&& gpg --export A4B469963BF863CC > /etc/apt/trusted.gpg.d/cuda.gpg
RUN apt-get update \
Expand All @@ -32,6 +34,8 @@ COPY --from=builder /var/lib/apt/lists* /var/lib/apt/lists/
COPY --from=builder /wheels /wheels
COPY . /video2x
WORKDIR /video2x
RUN gpg --keyserver=keyserver.ubuntu.com --receive-keys A4B469963BF863CC \
&& gpg --export A4B469963BF863CC > /etc/apt/trusted.gpg.d/cuda.gpg
RUN apt-get install -y --no-install-recommends \
python3-pip python3.9-dev \
python3-opencv python3-pil \
Expand Down

0 comments on commit 8b92bd5

Please sign in to comment.