setup-nvidia: Support aarch64 installer

Use `uname -m` to fetch the correct driver installer for aarch64 or x86_64.

Signed-off-by: Jeremi Piotrowski <jpiotrowski@microsoft.com>
This commit is contained in:
Jeremi Piotrowski 2025-02-14 14:55:06 +01:00
parent 209c3ab163
commit 3c4c7c6cb0

View File

@ -7,7 +7,7 @@ set -euo pipefail
[ -f /etc/flatcar/nvidia-metadata ] && . /etc/flatcar/nvidia-metadata
NVIDIA_DOWNLOAD_BASEURL="https://us.download.nvidia.com/${NVIDIA_PRODUCT_TYPE}/"
NVIDIA_DRIVER_BASENAME="NVIDIA-Linux-x86_64-${NVIDIA_DRIVER_VERSION}"
NVIDIA_DRIVER_BASENAME="NVIDIA-Linux-$(uname -m)-${NVIDIA_DRIVER_VERSION}"
NVIDIA_WORKDIR='nvidia-workdir'
FLATCAR_DEVELOPER_CONTAINER="flatcar_developer_container-${FLATCAR_RELEASE_VERSION}.bin"