mirror of
https://github.com/MacRimi/ProxMenux.git
synced 2026-02-19 08:56:23 +00:00
Create amd_gpu_tools.sh
This commit is contained in:
MacRimi
263
scripts/gpu_tpu/amd_gpu_tools.sh
Normal file
263
scripts/gpu_tpu/amd_gpu_tools.sh
Normal file
@@ -0,0 +1,263 @@
|
|||||||
|
#!/bin/bash
|
||||||
|
# ProxMenux - AMD GPU Tools Installer
|
||||||
|
# ============================================
|
||||||
|
# Author : MacRimi
|
||||||
|
# License : MIT
|
||||||
|
# Version : 1.0
|
||||||
|
# Last Updated: 29/01/2026
|
||||||
|
# ============================================
|
||||||
|
# Installs amdgpu_top for monitoring AMD GPUs
|
||||||
|
# https://github.com/Umio-Yasuno/amdgpu_top
|
||||||
|
|
||||||
|
SCRIPT_TITLE="AMD GPU Tools Installer for Proxmox VE"
|
||||||
|
|
||||||
|
LOCAL_SCRIPTS="c"
|
||||||
|
BASE_DIR="/usr/local/share/proxmenux"
|
||||||
|
UTILS_FILE="$BASE_DIR/utils.sh"
|
||||||
|
COMPONENTS_STATUS_FILE="$BASE_DIR/components_status.json"
|
||||||
|
LOG_FILE="/tmp/amd_gpu_tools_install.log"
|
||||||
|
|
||||||
|
export BASE_DIR
|
||||||
|
export COMPONENTS_STATUS_FILE
|
||||||
|
|
||||||
|
if [[ -f "$UTILS_FILE" ]]; then
|
||||||
|
source "$UTILS_FILE"
|
||||||
|
fi
|
||||||
|
|
||||||
|
if [[ ! -f "$COMPONENTS_STATUS_FILE" ]]; then
|
||||||
|
echo "{}" > "$COMPONENTS_STATUS_FILE"
|
||||||
|
fi
|
||||||
|
|
||||||
|
load_language 2>/dev/null || true
|
||||||
|
initialize_cache 2>/dev/null || true
|
||||||
|
|
||||||
|
# ==========================================================
|
||||||
|
# AMD GPU detection
|
||||||
|
# ==========================================================
|
||||||
|
detect_amd_gpus() {
|
||||||
|
local lspci_output
|
||||||
|
lspci_output=$(lspci | grep -iE "(AMD|ATI)" \
|
||||||
|
| grep -Ei "VGA compatible controller|3D controller|Display controller" || true)
|
||||||
|
|
||||||
|
if [[ -z "$lspci_output" ]]; then
|
||||||
|
AMD_GPU_PRESENT=false
|
||||||
|
DETECTED_GPUS_TEXT="No AMD GPU detected on this system."
|
||||||
|
else
|
||||||
|
AMD_GPU_PRESENT=true
|
||||||
|
DETECTED_GPUS_TEXT=""
|
||||||
|
local i=1
|
||||||
|
while IFS= read -r line; do
|
||||||
|
DETECTED_GPUS_TEXT+=" ${i}. ${line}\n"
|
||||||
|
((i++))
|
||||||
|
done <<< "$lspci_output"
|
||||||
|
fi
|
||||||
|
}
|
||||||
|
|
||||||
|
# ==========================================================
|
||||||
|
# Check if amdgpu_top is installed
|
||||||
|
# ==========================================================
|
||||||
|
check_amdgpu_top_installed() {
|
||||||
|
if command -v amdgpu_top >/dev/null 2>&1; then
|
||||||
|
AMDGPU_TOP_INSTALLED=true
|
||||||
|
AMDGPU_TOP_VERSION=$(amdgpu_top --version 2>/dev/null | head -n1 || echo "unknown")
|
||||||
|
else
|
||||||
|
AMDGPU_TOP_INSTALLED=false
|
||||||
|
AMDGPU_TOP_VERSION=""
|
||||||
|
fi
|
||||||
|
}
|
||||||
|
|
||||||
|
# ==========================================================
|
||||||
|
# Get latest amdgpu_top release from GitHub
|
||||||
|
# ==========================================================
|
||||||
|
get_latest_release() {
|
||||||
|
local api_url="https://api.github.com/repos/Umio-Yasuno/amdgpu_top/releases/latest"
|
||||||
|
|
||||||
|
LATEST_VERSION=$(curl -sL "$api_url" | grep '"tag_name"' | head -n1 | sed -E 's/.*"tag_name": *"([^"]+)".*/\1/')
|
||||||
|
|
||||||
|
if [[ -z "$LATEST_VERSION" ]]; then
|
||||||
|
msg_error "$(translate 'Failed to get latest version from GitHub')"
|
||||||
|
return 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
# Get the .deb download URL for amd64
|
||||||
|
DEB_URL=$(curl -sL "$api_url" | grep -oP '"browser_download_url":\s*"\K[^"]+amd64\.deb' | head -n1)
|
||||||
|
|
||||||
|
if [[ -z "$DEB_URL" ]]; then
|
||||||
|
msg_error "$(translate 'Failed to get .deb download URL')"
|
||||||
|
return 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
||||||
|
# ==========================================================
|
||||||
|
# Install dependencies
|
||||||
|
# ==========================================================
|
||||||
|
install_dependencies() {
|
||||||
|
msg_info "$(translate 'Installing required dependencies...')"
|
||||||
|
|
||||||
|
apt-get update -qq >>"$LOG_FILE" 2>&1
|
||||||
|
|
||||||
|
# Install libdrm packages required for amdgpu_top
|
||||||
|
if apt-get install -y libdrm-dev libdrm-amdgpu1 libdrm2 curl wget >>"$LOG_FILE" 2>&1; then
|
||||||
|
msg_ok "$(translate 'Dependencies installed successfully')"
|
||||||
|
return 0
|
||||||
|
else
|
||||||
|
msg_error "$(translate 'Failed to install dependencies')"
|
||||||
|
return 1
|
||||||
|
fi
|
||||||
|
}
|
||||||
|
|
||||||
|
# ==========================================================
|
||||||
|
# Install amdgpu_top
|
||||||
|
# ==========================================================
|
||||||
|
install_amdgpu_top() {
|
||||||
|
local tmp_dir="/tmp/amdgpu_top_install"
|
||||||
|
mkdir -p "$tmp_dir"
|
||||||
|
|
||||||
|
msg_info "$(translate 'Downloading amdgpu_top') ${LATEST_VERSION}..."
|
||||||
|
|
||||||
|
local deb_file="$tmp_dir/amdgpu_top.deb"
|
||||||
|
|
||||||
|
if ! wget -q -O "$deb_file" "$DEB_URL" >>"$LOG_FILE" 2>&1; then
|
||||||
|
msg_error "$(translate 'Failed to download amdgpu_top')"
|
||||||
|
rm -rf "$tmp_dir"
|
||||||
|
return 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
msg_ok "$(translate 'Downloaded amdgpu_top') ${LATEST_VERSION}"
|
||||||
|
|
||||||
|
msg_info "$(translate 'Installing amdgpu_top...')"
|
||||||
|
|
||||||
|
if ! dpkg -i "$deb_file" >>"$LOG_FILE" 2>&1; then
|
||||||
|
# Try to fix dependencies if dpkg failed
|
||||||
|
apt-get install -f -y >>"$LOG_FILE" 2>&1
|
||||||
|
if ! dpkg -i "$deb_file" >>"$LOG_FILE" 2>&1; then
|
||||||
|
msg_error "$(translate 'Failed to install amdgpu_top')"
|
||||||
|
rm -rf "$tmp_dir"
|
||||||
|
return 1
|
||||||
|
fi
|
||||||
|
fi
|
||||||
|
|
||||||
|
msg_ok "$(translate 'amdgpu_top installed successfully')"
|
||||||
|
|
||||||
|
# Clean up
|
||||||
|
rm -rf "$tmp_dir"
|
||||||
|
|
||||||
|
# Update component status
|
||||||
|
if type update_component_status &>/dev/null; then
|
||||||
|
update_component_status "amdgpu_top" "installed" "$LATEST_VERSION" "gpu" '{"source":"github"}'
|
||||||
|
fi
|
||||||
|
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
||||||
|
# ==========================================================
|
||||||
|
# Uninstall amdgpu_top
|
||||||
|
# ==========================================================
|
||||||
|
uninstall_amdgpu_top() {
|
||||||
|
msg_info "$(translate 'Uninstalling amdgpu_top...')"
|
||||||
|
|
||||||
|
if dpkg -r amdgpu-top >>"$LOG_FILE" 2>&1 || apt-get remove -y amdgpu-top >>"$LOG_FILE" 2>&1; then
|
||||||
|
msg_ok "$(translate 'amdgpu_top uninstalled successfully')"
|
||||||
|
|
||||||
|
if type update_component_status &>/dev/null; then
|
||||||
|
update_component_status "amdgpu_top" "uninstalled" "" "gpu" '{}'
|
||||||
|
fi
|
||||||
|
return 0
|
||||||
|
else
|
||||||
|
msg_error "$(translate 'Failed to uninstall amdgpu_top')"
|
||||||
|
return 1
|
||||||
|
fi
|
||||||
|
}
|
||||||
|
|
||||||
|
# ==========================================================
|
||||||
|
# Main execution
|
||||||
|
# ==========================================================
|
||||||
|
main() {
|
||||||
|
# Show ProxMenux logo and title
|
||||||
|
show_proxmenux_logo
|
||||||
|
msg_title "$(translate "$SCRIPT_TITLE")"
|
||||||
|
|
||||||
|
# Detect AMD GPUs
|
||||||
|
detect_amd_gpus
|
||||||
|
|
||||||
|
if ! $AMD_GPU_PRESENT; then
|
||||||
|
msg_warn "$(translate 'No AMD GPU detected on this system.')"
|
||||||
|
msg_info2 "$(translate 'This tool is designed for systems with AMD GPUs.')"
|
||||||
|
msg_info2 "$(translate 'You can still install amdgpu_top if needed.')"
|
||||||
|
echo ""
|
||||||
|
else
|
||||||
|
msg_ok "$(translate 'AMD GPU(s) detected:')"
|
||||||
|
echo -e "$DETECTED_GPUS_TEXT"
|
||||||
|
fi
|
||||||
|
|
||||||
|
# Check if already installed
|
||||||
|
check_amdgpu_top_installed
|
||||||
|
|
||||||
|
if $AMDGPU_TOP_INSTALLED; then
|
||||||
|
msg_ok "$(translate 'amdgpu_top is already installed:') $AMDGPU_TOP_VERSION"
|
||||||
|
|
||||||
|
# Check for updates
|
||||||
|
if get_latest_release; then
|
||||||
|
if [[ "$AMDGPU_TOP_VERSION" != *"$LATEST_VERSION"* ]]; then
|
||||||
|
msg_info2 "$(translate 'A newer version is available:') $LATEST_VERSION"
|
||||||
|
msg_info "$(translate 'Updating amdgpu_top...')"
|
||||||
|
|
||||||
|
if install_dependencies && install_amdgpu_top; then
|
||||||
|
msg_ok "$(translate 'amdgpu_top updated to') $LATEST_VERSION"
|
||||||
|
else
|
||||||
|
msg_error "$(translate 'Failed to update amdgpu_top')"
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
else
|
||||||
|
msg_ok "$(translate 'amdgpu_top is up to date')"
|
||||||
|
fi
|
||||||
|
fi
|
||||||
|
else
|
||||||
|
msg_info2 "$(translate 'amdgpu_top is not installed')"
|
||||||
|
msg_info "$(translate 'Starting installation...')"
|
||||||
|
|
||||||
|
# Get latest release info
|
||||||
|
if ! get_latest_release; then
|
||||||
|
msg_error "$(translate 'Failed to get release information from GitHub')"
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
msg_ok "$(translate 'Latest version:') $LATEST_VERSION"
|
||||||
|
|
||||||
|
# Install dependencies
|
||||||
|
if ! install_dependencies; then
|
||||||
|
msg_error "$(translate 'Failed to install dependencies')"
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
# Install amdgpu_top
|
||||||
|
if ! install_amdgpu_top; then
|
||||||
|
msg_error "$(translate 'Installation failed')"
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
fi
|
||||||
|
|
||||||
|
echo ""
|
||||||
|
msg_ok "$(translate 'AMD GPU Tools installation completed!')"
|
||||||
|
echo ""
|
||||||
|
msg_info2 "$(translate 'You can now monitor your AMD GPU using:')"
|
||||||
|
echo " amdgpu_top - $(translate 'TUI mode')"
|
||||||
|
echo " amdgpu_top --json - $(translate 'JSON output for scripts')"
|
||||||
|
echo " amdgpu_top --gui - $(translate 'GUI mode (if available)')"
|
||||||
|
echo ""
|
||||||
|
|
||||||
|
# In web mode, don't wait for user input
|
||||||
|
if ! is_web_mode 2>/dev/null; then
|
||||||
|
msg_success "$(translate 'Installation completed. Press Enter to continue...')"
|
||||||
|
read -r
|
||||||
|
else
|
||||||
|
msg_success "$(translate 'Installation completed.')"
|
||||||
|
fi
|
||||||
|
}
|
||||||
|
|
||||||
|
# Run main function
|
||||||
|
if [[ "${BASH_SOURCE[0]}" == "$0" ]]; then
|
||||||
|
main
|
||||||
|
fi
|
||||||
Reference in New Issue
Block a user