From d6e023572445f9b052e01f788a93f1037b5de879 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?H=C3=A9ctor=20Molinero=20Fern=C3=A1ndez?=
 <hector@molinero.dev>
Date: Wed, 5 May 2021 00:12:40 +0200
Subject: [PATCH] Add GPU devices groups to additional groups

---
 scripts/bin/container-init | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/scripts/bin/container-init b/scripts/bin/container-init
index bc9d76e..87b7681 100755
--- a/scripts/bin/container-init
+++ b/scripts/bin/container-init
@@ -5,6 +5,14 @@ set -eu
 # Remove leftover files
 find /tmp/ -mindepth 1 -delete
 
+# Add GPU devices groups to additional groups
+for dev in /dev/dri/*; do
+	[ -c "${dev:?}" ] || continue
+	gid=$(stat -c '%g' "${dev:?}")
+	uug=${UNPRIVILEGED_USER_GROUPS?}
+	UNPRIVILEGED_USER_GROUPS="${uug:+${uug:?},}${gid:?}"
+done
+
 # Create additional groups
 _IFS=${IFS}; IFS=,
 for gid in ${UNPRIVILEGED_USER_GROUPS?}; do