Michael Chus
559fc2961d
- 550.54.15 did not exist on NVIDIA CDN (404) - updated to 590.48.01 (latest stable, 396MB) - download sha256sum file first, verify installer before extracting - re-download if file is missing, empty, or sha256 mismatch Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
106 lines
4.0 KiB
Bash
106 lines
4.0 KiB
Bash
#!/bin/sh
|
|
# build-nvidia-module.sh — install NVIDIA proprietary driver into ISO overlay
|
|
#
|
|
# Downloads the official NVIDIA .run installer, extracts kernel modules and
|
|
# userspace tools (nvidia-smi, libnvidia-ml). Everything is proprietary NVIDIA.
|
|
#
|
|
# Output is cached in DIST_DIR/nvidia-<version>-<kver>/ so subsequent builds
|
|
# are instant unless NVIDIA_DRIVER_VERSION or kernel version changes.
|
|
#
|
|
# Output layout:
|
|
# $CACHE_DIR/modules/ — nvidia*.ko files
|
|
# $CACHE_DIR/bin/ — nvidia-smi, nvidia-debugdump
|
|
# $CACHE_DIR/lib/ — libnvidia-ml.so*, libcuda.so* (for nvidia-smi)
|
|
|
|
set -e
|
|
|
|
NVIDIA_VERSION="$1"
|
|
DIST_DIR="$2"
|
|
|
|
[ -n "$NVIDIA_VERSION" ] || { echo "usage: $0 <nvidia-version> <dist-dir>"; exit 1; }
|
|
[ -n "$DIST_DIR" ] || { echo "usage: $0 <nvidia-version> <dist-dir>"; exit 1; }
|
|
|
|
# Detect kernel version from installed headers
|
|
KVER=$(ls /usr/src/ 2>/dev/null | grep '^linux-headers-' | sed 's/linux-headers-//' | head -1)
|
|
if [ -z "$KVER" ]; then
|
|
echo "=== installing linux-lts-dev ==="
|
|
apk add --quiet linux-lts-dev
|
|
KVER=$(ls /usr/src/ | grep '^linux-headers-' | sed 's/linux-headers-//' | head -1)
|
|
fi
|
|
KDIR="/usr/src/linux-headers-${KVER}"
|
|
echo "=== NVIDIA ${NVIDIA_VERSION} (proprietary) for kernel ${KVER} ==="
|
|
|
|
CACHE_DIR="${DIST_DIR}/nvidia-${NVIDIA_VERSION}-${KVER}"
|
|
if [ -d "$CACHE_DIR/modules" ] && [ -f "$CACHE_DIR/bin/nvidia-smi" ]; then
|
|
echo "=== NVIDIA cached, skipping build ==="
|
|
echo "cache: $CACHE_DIR"
|
|
echo "modules: $(ls "$CACHE_DIR/modules/"*.ko 2>/dev/null | wc -l) .ko files"
|
|
exit 0
|
|
fi
|
|
|
|
# Install build dependencies
|
|
apk add --quiet gcc make perl linux-lts-dev wget
|
|
|
|
# Download official NVIDIA .run installer (proprietary) with sha256 verification
|
|
BASE_URL="https://download.nvidia.com/XFree86/Linux-x86_64/${NVIDIA_VERSION}"
|
|
RUN_FILE="/var/tmp/NVIDIA-Linux-x86_64-${NVIDIA_VERSION}.run"
|
|
SHA_FILE="/var/tmp/NVIDIA-Linux-x86_64-${NVIDIA_VERSION}.run.sha256sum"
|
|
|
|
verify_run() {
|
|
[ -s "$SHA_FILE" ] || return 1
|
|
[ -s "$RUN_FILE" ] || return 1
|
|
cd /var/tmp
|
|
sha256sum -c "$SHA_FILE" --status 2>/dev/null
|
|
}
|
|
|
|
if ! verify_run; then
|
|
rm -f "$RUN_FILE" "$SHA_FILE"
|
|
echo "=== downloading NVIDIA ${NVIDIA_VERSION} installer ==="
|
|
wget -q -O "$SHA_FILE" "${BASE_URL}/NVIDIA-Linux-x86_64-${NVIDIA_VERSION}.run.sha256sum"
|
|
echo "sha256: $(cat "$SHA_FILE")"
|
|
wget --show-progress -O "$RUN_FILE" "${BASE_URL}/NVIDIA-Linux-x86_64-${NVIDIA_VERSION}.run"
|
|
echo "=== verifying sha256 ==="
|
|
cd /var/tmp && sha256sum -c "$SHA_FILE" || { echo "ERROR: sha256 mismatch"; rm -f "$RUN_FILE"; exit 1; }
|
|
echo "sha256 OK"
|
|
else
|
|
echo "=== NVIDIA installer verified from cache ==="
|
|
fi
|
|
|
|
# Extract installer contents
|
|
echo "=== extracting installer ==="
|
|
chmod +x "$RUN_FILE"
|
|
EXTRACT_DIR="/var/tmp/nvidia-extract-${NVIDIA_VERSION}"
|
|
rm -rf "$EXTRACT_DIR"
|
|
"$RUN_FILE" --extract-only --target "$EXTRACT_DIR"
|
|
|
|
# Find kernel source directory (proprietary: kernel/, open: kernel-open/)
|
|
KERNEL_SRC=""
|
|
for d in "$EXTRACT_DIR/kernel" "$EXTRACT_DIR/kernel-modules-sources" "$EXTRACT_DIR/kernel-source"; do
|
|
[ -f "$d/Makefile" ] && KERNEL_SRC="$d" && break
|
|
done
|
|
[ -n "$KERNEL_SRC" ] || { echo "ERROR: kernel source dir not found in:"; ls "$EXTRACT_DIR/"; exit 1; }
|
|
echo "kernel source: $KERNEL_SRC"
|
|
|
|
# Build kernel modules from extracted source
|
|
echo "=== building kernel modules ($(nproc) cores) ==="
|
|
cd "$KERNEL_SRC"
|
|
make -j$(nproc) KERNEL_UNAME="$KVER" SYSSRC="$KDIR" modules 2>&1 | tail -5
|
|
|
|
# Collect outputs
|
|
mkdir -p "$CACHE_DIR/modules" "$CACHE_DIR/bin" "$CACHE_DIR/lib"
|
|
|
|
find "$EXTRACT_DIR/kernel" -name '*.ko' -exec cp {} "$CACHE_DIR/modules/" \;
|
|
for ko in "$CACHE_DIR/modules/"*.ko; do
|
|
strip --strip-debug "$ko" 2>/dev/null || true
|
|
done
|
|
|
|
cp "$EXTRACT_DIR/nvidia-smi" "$CACHE_DIR/bin/"
|
|
cp "$EXTRACT_DIR/libnvidia-ml.so."* "$CACHE_DIR/lib/" 2>/dev/null || true
|
|
# libcuda stub needed by nvidia-smi at runtime
|
|
cp "$EXTRACT_DIR/libcuda.so."* "$CACHE_DIR/lib/" 2>/dev/null || true
|
|
|
|
echo "=== NVIDIA build complete ==="
|
|
echo "cache: $CACHE_DIR"
|
|
echo "modules: $(ls "$CACHE_DIR/modules/"*.ko | wc -l) .ko files"
|
|
ls -lh "$CACHE_DIR/bin/"
|