Skip to content

feat: add --nvml flag to install scripts for NVML GPU build #7

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
5o1 merged 2 commits into from
Jul 8, 2026
Merged

feat: add --nvml flag to install scripts for NVML GPU build #7

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
491f2c3
feat: add --nvml flag to install scripts for NVML GPU build
5o1 Jul 8, 2026
2a940c5
Merge branch 'main' into feat/nvml-build-option
5o1 Jul 8, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
18 changes: 14 additions & 4 deletions scripts/install.sh
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -10,6 +10,8 @@
# sudo ./scripts/install.sh --port 9090 # install with custom port
# ./scripts/install.sh -h # show usage

# sudo ./scripts/install.sh --nvml # build with NVML GPU support
# sudo ./scripts/install.sh --nvml --port 9090
set -euo pipefail

HERE="$(cd "$(dirname "$0")/.." && pwd)"
Expand All @@ -28,6 +30,7 @@ CONFIG_EXAMPLE="${HERE}/config.example.yaml"
MONITOR_USER="root"
DO_START=1
PORT=""
NVML_BUILD=0

# ---------------------------------------------------------------------------
# Colors
Expand All @@ -53,6 +56,7 @@ Build and install workstation-probe (monitor) as a system service.
Requires root privileges.

Options:
--nvml Build with NVML GPU collector (requires libnvidia-ml).
-h Show this help and exit.
--port N Set the server listen port (default: 19090 from config.example.yaml).
--no-start Install the binary, config, and service, but do not start the service.
Expand All @@ -78,6 +82,7 @@ fi
while [ $# -gt 0 ]; do
case "$1" in
-h) usage ;;
--nvml) NVML_BUILD=1 ;;
--port)
shift
if [ $# -eq 0 ]; then
Expand Down Expand Up @@ -117,10 +122,15 @@ fi

# ---------------------------------------------------------------------------
# Build
# ---------------------------------------------------------------------------
info "Building binary..."
if ! make build; then
fatal "Build failed. Check the Go toolchain and try again."
if [ "$NVML_BUILD" -eq 1 ]; then
info "Building with NVML GPU support..."
if ! make build-nvml; then
fatal "Build failed. Check the Go toolchain, libnvidia-ml, and try again."
fi
else
if ! make build; then
fatal "Build failed. Check the Go toolchain and try again."
fi
fi
info "Build succeeded."

Expand Down
18 changes: 15 additions & 3 deletions scripts/rootless-install.sh
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -10,6 +10,8 @@
# ./scripts/rootless-install.sh --no-start # install without starting
# ./scripts/rootless-install.sh --port 9090 # install with custom port
# ./scripts/rootless-install.sh -h # show usage
# ./scripts/rootless-install.sh --nvml # build with NVML GPU support
# ./scripts/rootless-install.sh --nvml --port 9090

set -euo pipefail

Expand All @@ -28,6 +30,7 @@ SERVICE_FILE="${SYSTEMD_USER_DIR}/${BIN_NAME}.service"
CONFIG_EXAMPLE="${HERE}/config.example.yaml"
DO_START=1
PORT=""
NVML_BUILD=0

# ---------------------------------------------------------------------------
# Colors
Expand All @@ -53,6 +56,7 @@ Build and install workstation-probe (monitor) as a user service (rootless).
No root privileges required.

Options:
--nvml Build with NVML GPU collector (requires libnvidia-ml).
-h Show this help and exit.
--port N Set the server listen port (default: 19090 from config.example.yaml).
--no-start Install the binary, config, and service, but do not start the service.
Expand All @@ -71,6 +75,7 @@ EOF
while [ $# -gt 0 ]; do
case "$1" in
-h) usage ;;
--nvml) NVML_BUILD=1 ;;
--port)
shift
if [ $# -eq 0 ]; then
Expand Down Expand Up @@ -112,11 +117,18 @@ fi
# ---------------------------------------------------------------------------
# Build
# ---------------------------------------------------------------------------
info "Building binary..."
if ! make build; then
fatal "Build failed. Check the Go toolchain and try again."
if [ "$NVML_BUILD" -eq 1 ]; then
info "Building with NVML GPU support..."
if ! make build-nvml; then
fatal "Build failed. Check the Go toolchain, libnvidia-ml, and try again."
fi
else
if ! make build; then
fatal "Build failed. Check the Go toolchain and try again."
fi
fi
info "Build succeeded."
info "Build succeeded."

# ---------------------------------------------------------------------------
# Install binary
Expand Down
Loading