mirror of
https://github.com/TrezOne/docker-mods-uptime-kuma-timeout-fix.git
synced 2026-07-19 09:53:02 -04:00
aptalca
78 lines
2.9 KiB
Plaintext
Executable File
78 lines
2.9 KiB
Plaintext
Executable File
#!/usr/bin/with-contenv bash
|
|
|
|
AMDGPU_RELEASE=${AMDGPU_RELEASE:-5.4.1}
|
|
|
|
# Determine if setup is needed
|
|
if [ ! -f "/usr/bin/apt" ]; then
|
|
echo "**** Image is not Ubuntu, skipping opencl-intel install ****"
|
|
exit 0
|
|
fi
|
|
|
|
if ! dpkg -l | grep gnupg > /dev/null; then
|
|
apt-get update && apt-get install -y gnupg
|
|
fi
|
|
|
|
if [ ! -f "/etc/apt/sources.list.d/kisak-mesa.list" ]; then
|
|
echo "**** Adding kisak-mesa repo ****"
|
|
apt-key adv --keyserver keyserver.ubuntu.com --recv-keys F63F0F2B90935439
|
|
source /etc/lsb-release
|
|
echo "deb http://ppa.launchpad.net/kisak/kisak-mesa/ubuntu ${DISTRIB_CODENAME} main" > /etc/apt/sources.list.d/kisak-mesa.list
|
|
fi
|
|
|
|
if [ ! -f "/etc/apt/sources.list.d/amdgpu-focal.list" ]; then
|
|
echo "**** Adding amdgpu repo ****"
|
|
curl -sL --retry 3 https://repo.radeon.com/rocm/rocm.gpg.key | apt-key add -
|
|
echo "deb [arch=amd64] https://repo.radeon.com/rocm/apt/${AMDGPU_RELEASE} ubuntu main" > /etc/apt/sources.list.d/amdgpu-focal.list
|
|
fi
|
|
|
|
|
|
pkgs='mesa-vdpau-drivers mesa-va-drivers mesa-vdpau-drivers libdrm-radeon1 rocm-opencl-runtime'
|
|
|
|
install=false
|
|
for pkg in $pkgs; do
|
|
status="$(dpkg-query -W --showformat='${db:Status-Status}' "$pkg" 2>&1)"
|
|
if [ ! $? = 0 ] || [ ! "$status" = installed ]; then
|
|
install=true
|
|
break
|
|
fi
|
|
done
|
|
|
|
if "$install"; then
|
|
echo "**** Adding mesa and rocm-opencl-runtime to package install list ****"
|
|
echo "$pkgs" >> /mod-repo-packages-to-install.list
|
|
fi
|
|
|
|
FILES=$(find /dev/kfd -type c -print 2>/dev/null)
|
|
|
|
for i in $FILES
|
|
do
|
|
VIDEO_GID=$(stat -c '%g' "${i}")
|
|
VIDEO_UID=$(stat -c '%u' "${i}")
|
|
# check if user matches device
|
|
if id -u abc | grep -qw "${VIDEO_UID}"; then
|
|
echo "**** permissions for ${i} are good ****"
|
|
else
|
|
# check if group matches and that device has group rw
|
|
if id -G abc | grep -qw "${VIDEO_GID}" && [ $(stat -c '%A' "${i}" | cut -b 5,6) = "rw" ]; then
|
|
echo "**** permissions for ${i} are good ****"
|
|
# check if device needs to be added to video group
|
|
elif ! id -G abc | grep -qw "${VIDEO_GID}"; then
|
|
# check if video group needs to be created
|
|
VIDEO_NAME=$(getent group "${VIDEO_GID}" | awk -F: '{print $1}')
|
|
if [ -z "${VIDEO_NAME}" ]; then
|
|
VIDEO_NAME="video$(head /dev/urandom | tr -dc 'a-z0-9' | head -c4)"
|
|
groupadd "${VIDEO_NAME}"
|
|
groupmod -g "${VIDEO_GID}" "${VIDEO_NAME}"
|
|
echo "**** creating video group ${VIDEO_NAME} with id ${VIDEO_GID} ****"
|
|
fi
|
|
echo "**** adding ${i} to video group ${VIDEO_NAME} with id ${VIDEO_GID} ****"
|
|
usermod -a -G "${VIDEO_NAME}" abc
|
|
fi
|
|
# check if device has group rw
|
|
if [ $(stat -c '%A' "${i}" | cut -b 5,6) != "rw" ]; then
|
|
echo -e "**** The device ${i} does not have group read/write permissions, attempting to fix inside the container. ****"
|
|
chmod g+rw "${i}"
|
|
fi
|
|
fi
|
|
done
|