From 23eecb71a8a3b4acdccc5a9127e82642855c5bf4 Mon Sep 17 00:00:00 2001
From: Julio Perez <37191411+jperez999@users.noreply.github.com>
Date: Mon, 9 May 2022 10:15:44 -0400
Subject: [PATCH] Add key update mechanism for ci dockerfile (#292)

* add entrypoint to all containers

* remove -e for pip installs

* add key update mechanism to ci dockerfile
---
 ci/dockerfile.ci | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/ci/dockerfile.ci b/ci/dockerfile.ci
index 3643c883b..568e34a76 100644
--- a/ci/dockerfile.ci
+++ b/ci/dockerfile.ci
@@ -30,6 +30,15 @@ RUN ln -s /usr/local/cuda/lib64/stubs/libcuda.so /usr/local/cuda/lib64/stubs/lib
 
 # Install packages
 ENV DEBIAN_FRONTEND=noninteractive
+
+RUN apt update -y --fix-missing && \
+    apt install -y --no-install-recommends software-properties-common && \
+    wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin && \
+    mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600 && \
+    apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/3bf863cc.pub && \
+    add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/ /"
+
+
 RUN apt update -y --fix-missing && \
     apt install -y --no-install-recommends software-properties-common && \
     apt-get install -y --no-install-recommends \