From 954c4f743f6cbd0af66503382d8b7f7be24f3d0d Mon Sep 17 00:00:00 2001
From: Hemal Shah <77975110+hemalshahNV@users.noreply.github.com>
Date: Thu, 12 May 2022 12:42:08 -0700
Subject: [PATCH] Update Dockerfile x86_64 to install nvidia apt-get keys

---
 docker/Dockerfile.x86_64.base | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/docker/Dockerfile.x86_64.base b/docker/Dockerfile.x86_64.base
index f36478d9..ef0d7921 100644
--- a/docker/Dockerfile.x86_64.base
+++ b/docker/Dockerfile.x86_64.base
@@ -14,6 +14,13 @@ FROM ${BASE_IMAGE}
 # disable terminal interaction for apt
 ENV DEBIAN_FRONTEND=noninteractive
 
+# NVIDIA repository keys: https://developer.nvidia.com/blog/updating-the-cuda-linux-gpg-repository-key/
+RUN apt-key del 7fa2af80 && mkdir -p /tmp && cd /tmp \
+        && wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-keyring_1.0-1_all.deb \
+        && dpkg -i cuda-keyring_1.0-1_all.deb \
+        && rm cuda-keyring_1.0-1_all.deb \
+        && add-apt-repository --remove 'deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/ /'
+
 # Fundamentals
 RUN apt-get update && apt-get install -y \
         bash-completion \