photoprism/scripts/dist/install-gpu.sh
Michael Mayer 8c589e3649 Videos: Improve Nvidia hardware transcoding support #2125
- successfully tested with NVIDIA Quadro P620 and driver v470.103.01
- the host Linux kernel should run the same driver version Make sure to
- driver names in PHOTOPRISM_FFMPEG_ENCODER have been simplified
- share /dev/nvidia* as shown in our new docker-compose.yml example
2022-03-23 13:27:25 +01:00

51 lines
1.1 KiB
Bash
Executable file

#!/bin/bash
PATH="/usr/local/sbin:/usr/sbin:/sbin:/usr/local/bin:/usr/bin:/bin:/scripts"
# abort if not executed as root
if [[ $(id -u) != "0" ]]; then
echo "Error: Run ${0##*/} as root" 1>&2
exit 1
fi
set -e
SYSTEM_ARCH=$("$(dirname "$0")/arch.sh")
DESTARCH=${DESTARCH:-$SYSTEM_ARCH}
TMPDIR=${TMPDIR:-/tmp}
. /etc/os-release
if [[ $DESTARCH != "amd64" ]]; then
echo "Installing GPU drivers for ${DESTARCH} is not supported yet."
exit
fi
apt-get update
apt-get -qq upgrade
apt-get -qq install lshw jq
# shellcheck disable=SC2207
GPU_DETECTED=($(lshw -c display -json 2>/dev/null | jq -r '.[].configuration.driver'))
# shellcheck disable=SC2068
for t in ${GPU_DETECTED[@]}; do
case $t in
i915)
apt-get -qq install intel-opencl-icd intel-media-va-driver-non-free i965-va-driver-shaders libmfx1 libva2 vainfo libva-wayland2
;;
nvidia)
apt-get -qq install nvidia-opencl-icd nvidia-vdpau-driver nvidia-driver-libs nvidia-kernel-dkms libva2 vainfo libva-wayland2 libnvidia-encode1
;;
null)
# ignore
;;
*)
echo "Unsupported GPU: \"$t\"";
;;
esac
done
echo "Done."