From 604efebf2f784c407edff09a02d33a181fe03635 Mon Sep 17 00:00:00 2001
From: Huamin Chen <hchen@redhat.com>
Date: Wed, 22 Sep 2021 11:11:16 -0400
Subject: [PATCH 1/2] install nvidia container kit in aio image

---
 hack/all-in-one/Dockerfile | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/hack/all-in-one/Dockerfile b/hack/all-in-one/Dockerfile
index c0ef48536d..dd799e2215 100644
--- a/hack/all-in-one/Dockerfile
+++ b/hack/all-in-one/Dockerfile
@@ -30,4 +30,9 @@ RUN export ARCH=$(uname -m |sed -e "s/x86_64/amd64/" |sed -e "s/aarch64/arm64/")
     systemctl enable microshift.service && \
     systemctl enable crio 
 
+RUN curl -s -L https://nvidia.github.io/nvidia-docker/rhel8.3/nvidia-docker.repo | tee /etc/yum.repos.d/nvidia-docker.repo && \
+    dnf install nvidia-container-toolkit -y
+
+ADD nvidia-config.toml /etc/nvidia-container-runtime/config.toml
+
 CMD [ "/sbin/init" ]

From a86a1537102e41aa426e9b9164b5496e985b1161 Mon Sep 17 00:00:00 2001
From: Huamin Chen <hchen@redhat.com>
Date: Wed, 22 Sep 2021 11:13:20 -0400
Subject: [PATCH 2/2] add nvidia container config

---
 hack/all-in-one/nvidia-config.toml | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)
 create mode 100644 hack/all-in-one/nvidia-config.toml

diff --git a/hack/all-in-one/nvidia-config.toml b/hack/all-in-one/nvidia-config.toml
new file mode 100644
index 0000000000..2424f88578
--- /dev/null
+++ b/hack/all-in-one/nvidia-config.toml
@@ -0,0 +1,18 @@
+disable-require = false
+#swarm-resource = "DOCKER_RESOURCE_GPU"
+#accept-nvidia-visible-devices-envvar-when-unprivileged = true
+#accept-nvidia-visible-devices-as-volume-mounts = false
+
+[nvidia-container-cli]
+#root = "/run/nvidia/driver"
+#path = "/usr/bin/nvidia-container-cli"
+environment = []
+debug = "/var/log/nvidia-container-toolkit.log"
+#ldcache = "/etc/ld.so.cache"
+load-kmods = false
+no-cgroups = false
+#user = "root:video"
+ldconfig = "@/sbin/ldconfig"
+
+[nvidia-container-runtime]
+debug = "/var/log/nvidia-container-runtime.log"