diff --git a/scripts/tools/gpu-amd.func b/scripts/tools/gpu-amd.func new file mode 100644 index 0000000..412d0a0 --- /dev/null +++ b/scripts/tools/gpu-amd.func @@ -0,0 +1,61 @@ +#!/usr/bin/env bash + +# AMD GPU Helper Functions for Proxmox LXC / ROCm passthrough +# Author: CanbiZ +# License: MIT + +set -euo pipefail + +function exit_script() { + printf "⚠️ User exited script\n" + exit 0 +} + +function msg() { + local type="$1" + shift + case "$type" in + info) printf " \033[36m➤\033[0m %s\n" "$@" ;; + ok) printf " \033[32m✔\033[0m %s\n" "$@" ;; + warn) printf " \033[33m⚠\033[0m %s\n" "$@" >&2 ;; + err) printf " \033[31m✘\033[0m %s\n" "$@" >&2 ;; + esac +} + +function amd_gpu_available() { + lspci | grep -qi 'VGA.*AMD' && [[ -e /dev/kfd ]] +} + +function passthrough_amd_to_lxc() { + local ctid="$1" + local conf="/etc/pve/lxc/${ctid}.conf" + + if ! amd_gpu_available; then + msg warn "No AMD GPU with ROCm support detected" + return 1 + fi + + { + echo "# AMD ROCm GPU" + echo "lxc.cgroup2.devices.allow: c 226:* rwm" + echo "lxc.cgroup2.devices.allow: c 238:* rwm" + echo "lxc.mount.entry: /dev/kfd dev/kfd none bind,optional,create=file" + echo "lxc.mount.entry: /dev/dri dev/dri none bind,optional,create=dir" + } >>"$conf" + + msg ok "AMD ROCm passthrough applied to CT $ctid" + return 0 +} + +function install_amd_tools_in_ct() { + local ctid="$1" + msg info "Installing AMD GPU tools in CT $ctid..." + + pct exec "$ctid" -- bash -c \ + "apt-get update && \ + apt-get install -y rocm-smi rocm-utils && \ + adduser \$(id -un 0) video && \ + adduser \$(id -un 0) render" >/dev/null 2>&1 || true + + msg ok "Installed ROCm tools inside CT $ctid" +} diff --git a/scripts/tools/gpu-intel.func b/scripts/tools/gpu-intel.func new file mode 100644 index 0000000..43e82e1 --- /dev/null +++ b/scripts/tools/gpu-intel.func @@ -0,0 +1,62 @@ +#!/usr/bin/env bash + +# Intel GPU Helper Functions for Proxmox LXC / VAAPI passthrough +# Author: CanbiZ +# License: MIT + +set -euo pipefail + +function exit_script() { + printf "⚠️ User exited script\n" + exit 0 +} + +function msg() { + local type="$1" + shift + case "$type" in + info) printf " \033[36m➤\033[0m %s\n" "$@" ;; + ok) printf " \033[32m✔\033[0m %s\n" "$@" ;; + warn) printf " \033[33m⚠\033[0m %s\n" "$@" >&2 ;; + err) printf " \033[31m✘\033[0m %s\n" "$@" >&2 ;; + esac +} + +function intel_gpu_available() { + [[ -e /dev/dri/renderD128 ]] && lspci | grep -qi 'VGA.*Intel' +} + +function passthrough_intel_to_lxc() { + local ctid="$1" + local conf="/etc/pve/lxc/${ctid}.conf" + + if ! intel_gpu_available; then + msg warn "No Intel iGPU detected on host" + return 1 + fi + + { + echo "# Intel iGPU (VAAPI)" + echo "lxc.cgroup2.devices.allow: c 226:* rwm" + echo "lxc.cgroup2.devices.allow: c 29:0 rwm" + echo "lxc.mount.entry: /dev/fb0 dev/fb0 none bind,optional,create=file" + echo "lxc.mount.entry: /dev/dri dev/dri none bind,optional,create=dir" + echo "lxc.mount.entry: /dev/dri/renderD128 dev/dri/renderD128 none bind,optional,create=file" + } >>"$conf" + + msg ok "Intel VAAPI passthrough applied to CT $ctid" + return 0 +} + +function install_intel_tools_in_ct() { + local ctid="$1" + msg info "Installing Intel tools in CT $ctid..." + + pct exec "$ctid" -- bash -c \ + "apt-get update && \ + apt-get install -y va-driver-all vainfo intel-gpu-tools ocl-icd-libopencl1 intel-opencl-icd && \ + adduser \$(id -un 0) video && \ + adduser \$(id -un 0) render" >/dev/null 2>&1 + + msg ok "Installed VAAPI & GPU tools inside CT $ctid" +} diff --git a/scripts/tools/hw-acceleration.sh b/scripts/tools/hw-acceleration.sh index 3e77797..2d412c6 100644 --- a/scripts/tools/hw-acceleration.sh +++ b/scripts/tools/hw-acceleration.sh @@ -23,11 +23,19 @@ # - Container driver installation via APT # - User group assignments (video/render) # - Interactive menu system via whiptail - -#!/usr/bin/env bash +# +# Proxmox LXC Hardware Passthrough & GPU Acceleration Setup +# https://github.com/community-scripts/ProxmoxVED set -euo pipefail +TEMP_DIR=$(mktemp -d) +trap 'rm -rf $TEMP_DIR' EXIT + +source <(wget -qO- https://github.com/community-scripts/ProxmoxVED/raw/main/scripts/tools/gpu-nvidia.func) +source <(wget -qO- https://github.com/community-scripts/ProxmoxVED/raw/main/scripts/tools/gpu-intel.func) +source <(wget -qO- https://github.com/community-scripts/ProxmoxVED/raw/main/scripts/tools/gpu-amd.func) + function header_info() { clear cat <<"EOF" @@ -45,66 +53,82 @@ function msg() { local type="$1" shift case "$type" in - info) printf " \033[36m➤\033[0m %s\n" "$*" ;; - ok) printf " \033[32m✔\033[0m %s\n" "$*" ;; - warn) printf " \033[33m⚠\033[0m %s\n" "$*" >&2 ;; - err) printf " \033[31m✘\033[0m %s\n" "$*" >&2 ;; + info) printf " \033[36m➤\033[0m %s\n" "$@" ;; + ok) printf " \033[32m✔\033[0m %s\n" "$@" ;; + warn) printf " \033[33m⚠\033[0m %s\n" "$@" >&2 ;; + err) printf " \033[31m✘\033[0m %s\n" "$@" >&2 ;; esac } -function detect_features() { - AVAILABLE_FEATURES=() - [[ -e /dev/ttyUSB0 || -e /dev/ttyACM0 ]] && AVAILABLE_FEATURES+=("usb" "USB Passthrough" OFF) - [[ -e /dev/dri/renderD128 ]] && AVAILABLE_FEATURES+=("intel" "Intel VAAPI GPU" OFF) - [[ -e /dev/nvidia0 ]] && AVAILABLE_FEATURES+=("nvidia" "NVIDIA GPU" OFF) - [[ -e /dev/kfd ]] && AVAILABLE_FEATURES+=("amd" "AMD GPU (ROCm)" OFF) +function prompt_features() { + local features=() + printf "\nAvailable features:\n" + if [[ -e /dev/ttyUSB0 || -e /dev/ttyACM0 ]]; then + echo " [1] USB Passthrough" + features+=("usb") + fi + if [[ -e /dev/dri/renderD128 ]]; then + echo " [2] Intel iGPU (VAAPI)" + features+=("intel") + fi + if [[ -e /dev/nvidia0 ]]; then + echo " [3] NVIDIA GPU" + features+=("nvidia") + fi + if [[ -e /dev/kfd ]]; then + echo " [4] AMD GPU (ROCm)" + features+=("amd") + fi - if [[ ${#AVAILABLE_FEATURES[@]} -eq 0 ]]; then - msg warn "No supported hardware found on host system." + if [[ ${#features[@]} -eq 0 ]]; then + msg err "No supported hardware found on host." + exit 1 + fi + + echo + read -rp "Enter number(s) separated by space (e.g. 1 3): " choices + SELECTED_FEATURES=() + for i in $choices; do + case "$i" in + 1) SELECTED_FEATURES+=("usb") ;; + 2) SELECTED_FEATURES+=("intel") ;; + 3) SELECTED_FEATURES+=("nvidia") ;; + 4) SELECTED_FEATURES+=("amd") ;; + esac + done + + if [[ ${#SELECTED_FEATURES[@]} -eq 0 ]]; then + msg warn "No valid feature selected." exit 1 fi } -function select_hw_features() { - SELECTED_FEATURES=$(whiptail --title "Hardware Options" --checklist \ - "Select hardware features to passthrough:" 20 60 10 \ - "${AVAILABLE_FEATURES[@]}" 3>&1 1>&2 2>&3 | tr -d '"') || exit 1 - - [[ -z "$SELECTED_FEATURES" ]] && { - msg warn "No passthrough options selected." - exit 1 - } -} - -function select_lxc_targets() { - local opts=() - while IFS= read -r line; do - local id name conf - id=$(awk '{print $1}' <<<"$line") - name=$(awk '{print $2}' <<<"$line") - conf="/etc/pve/lxc/${id}.conf" - [[ -f "$conf" ]] && opts+=("$id" "$name (CTID: $id)" OFF) - done < <(pct list | tail -n +2) - - if [[ ${#opts[@]} -eq 0 ]]; then - msg warn "No containers found. Make sure you have running LXCs." +function select_lxc_cts() { + mapfile -t containers < <(pct list | awk 'NR>1 {print $1 "|" $2}') + if [[ ${#containers[@]} -eq 0 ]]; then + msg warn "No LXC containers found." exit 1 fi - SELECTED_CTIDS=$(whiptail --title "Select LXC Containers" --checklist \ - "Choose container(s) to apply passthrough:" 20 60 10 \ - "${opts[@]}" 3>&1 1>&2 2>&3 | tr -d '"') || exit 1 + echo + echo "Available Containers:" + for entry in "${containers[@]}"; do + ctid="${entry%%|*}" + name="${entry##*|}" + echo " [$ctid] $name" + done - [[ -z "$SELECTED_CTIDS" ]] && { + echo + read -rp "Enter container ID(s) separated by space: " SELECTED_CTIDS + if [[ -z "$SELECTED_CTIDS" ]]; then msg warn "No containers selected." exit 1 - } + fi } -function apply_usb() { +function apply_usb_passthrough() { local conf="$1" - grep -q "ttyUSB\|ttyACM" <<<"$(ls /dev 2>/dev/null)" || return 1 - grep -q "ttyUSB" "$conf" 2>/dev/null && return 0 + grep -q "ttyUSB" "$conf" 2>/dev/null && return cat <>"$conf" # USB Passthrough lxc.cgroup2.devices.allow: a @@ -119,116 +143,59 @@ lxc.mount.entry: /dev/ttyACM1 dev/ttyACM1 none bind,optional,create= EOF } -function apply_intel() { - local conf="$1" - [[ -e /dev/dri/renderD128 ]] || return 1 - grep -q "renderD128" "$conf" 2>/dev/null && return 0 - cat <>"$conf" -# Intel VAAPI -lxc.cgroup2.devices.allow: c 226:* rwm -lxc.cgroup2.devices.allow: c 29:0 rwm -lxc.mount.entry: /dev/fb0 dev/fb0 none bind,optional,create=file -lxc.mount.entry: /dev/dri dev/dri none bind,optional,create=dir -lxc.mount.entry: /dev/dri/renderD128 dev/dri/renderD128 none bind,optional,create=file -EOF -} - -function apply_nvidia() { - local conf="$1" - [[ -e /dev/nvidia0 ]] || return 1 - grep -q "nvidia0" "$conf" 2>/dev/null && return 0 - cat <>"$conf" -# NVIDIA GPU -lxc.cgroup2.devices.allow: c 195:* rwm -lxc.mount.entry: /dev/nvidia0 dev/nvidia0 none bind,optional,create=file -lxc.mount.entry: /dev/nvidiactl dev/nvidiactl none bind,optional,create=file -lxc.mount.entry: /dev/nvidia-uvm dev/nvidia-uvm none bind,optional,create=file -lxc.mount.entry: /dev/nvidia-uvm-tools dev/nvidia-uvm-tools none bind,optional,create=file -EOF -} - -function apply_amd() { - local conf="$1" - [[ -e /dev/kfd ]] || return 1 - grep -q "/dev/kfd" "$conf" 2>/dev/null && return 0 - cat <>"$conf" -# AMD ROCm GPU -lxc.cgroup2.devices.allow: c 226:* rwm -lxc.cgroup2.devices.allow: c 238:* rwm -lxc.mount.entry: /dev/kfd dev/kfd none bind,optional,create=file -lxc.mount.entry: /dev/dri dev/dri none bind,optional,create=dir -EOF -} - -function install_drivers() { - local ctid="$1" - for opt in $SELECTED_FEATURES; do - case "$opt" in - intel) - msg info "Installing Intel drivers/tools in CT $ctid..." - pct exec "$ctid" -- bash -c " - apt-get update -qq - DEBIAN_FRONTEND=noninteractive apt-get install -y \ - va-driver-all vainfo intel-gpu-tools \ - ocl-icd-libopencl1 intel-opencl-icd >/dev/null - adduser root video >/dev/null 2>&1 || true - adduser root render >/dev/null 2>&1 || true - " - ;; - nvidia) - msg info "Installing NVIDIA container tools in CT $ctid..." - pct exec "$ctid" -- bash -c " - apt-get update -qq - DEBIAN_FRONTEND=noninteractive apt-get install -y \ - nvidia-container-runtime nvidia-utils-525 >/dev/null 2>&1 || true - " - ;; - amd) - msg info "Installing AMD ROCm tools in CT $ctid..." - pct exec "$ctid" -- bash -c " - apt-get update -qq - DEBIAN_FRONTEND=noninteractive apt-get install -y \ - rocm-smi rocm-utils >/dev/null 2>&1 || true - " - ;; - esac - done -} - function main() { header_info - detect_features - select_hw_features - select_lxc_targets + prompt_features + select_lxc_cts local updated_cts=() + for ctid in $SELECTED_CTIDS; do local conf="/etc/pve/lxc/${ctid}.conf" local updated=0 - for opt in $SELECTED_FEATURES; do - case "$opt" in - usb) apply_usb "$conf" && updated=1 ;; - intel) apply_intel "$conf" && updated=1 ;; - nvidia) apply_nvidia "$conf" && updated=1 ;; - amd) apply_amd "$conf" && updated=1 ;; + + for feature in "${SELECTED_FEATURES[@]}"; do + case "$feature" in + usb) + msg info "Applying USB passthrough to CT $ctid..." + apply_usb_passthrough "$conf" && updated=1 + ;; + intel) + msg info "Applying Intel VAAPI passthrough to CT $ctid..." + passthrough_intel_to_lxc "$ctid" && install_intel_tools_in_ct "$ctid" && updated=1 + ;; + amd) + msg info "Applying AMD GPU passthrough to CT $ctid..." + passthrough_amd_to_lxc "$ctid" && install_amd_tools_in_ct "$ctid" && updated=1 + ;; + nvidia) + msg info "Checking NVIDIA GPU on host..." + check_nvidia_driver_status && check_cuda_version + gpu_minor=$(select_nvidia_gpu) || continue + passthrough_nvidia_to_lxc "$ctid" "$gpu_minor" && updated=1 + ;; esac done - [[ "$updated" -eq 1 ]] && updated_cts+=("$ctid") - install_drivers "$ctid" + + if [[ "$updated" -eq 1 ]]; then + updated_cts+=("$ctid") + fi done + echo if [[ ${#updated_cts[@]} -gt 0 ]]; then - msg ok "Hardware passthrough updated in: ${updated_cts[*]}" - if whiptail --yesno "Restart updated container(s)?\n${updated_cts[*]}" 10 60; then + msg ok "Updated: ${updated_cts[*]}" + read -rp "Restart updated container(s)? [y/N]: " restart + if [[ "${restart,,}" == "y" ]]; then for ctid in "${updated_cts[@]}"; do pct restart "$ctid" + msg ok "Restarted container $ctid" done - msg ok "Containers restarted: ${updated_cts[*]}" else - msg info "Please restart the container(s) manually." + msg info "Manual restart required for: ${updated_cts[*]}" fi else - msg warn "No passthrough or driver changes were applied." + msg warn "No passthrough applied." fi }