github/kata-containers
1
0
Fork 2
mirror of https://github.com/kata-containers/kata-containers.git synced 2025-09-18 07:18:27 +00:00
Code Issues Projects Releases Wiki Activity

gpu: Add driver version selection

Browse Source 
Besides latest and lts options add an option to specify
the exact driver version.

Signed-off-by: Zvonko Kaiser <zkaiser@nvidia.com>
This commit is contained in:
Zvonko Kaiser
2025-01-16 23:31:20 +00:00
parent 311c3638c6
commit f153229865
3 changed files with 71 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
tools/osbuilder/rootfs-builder/nvidia/nvidia_rootfs.sh
View File

@@ -17,6 +17,7 @@ DEBUG=""
setup_nvidia-nvrc() {
local TARGET="nvidia-nvrc"
local TARGET_VERSION="main"
local PROJECT="nvrc"
local TARGET_BUILD_DIR="${BUILD_DIR}/${TARGET}/builddir"
local TARGET_DEST_DIR="${BUILD_DIR}/${TARGET}/destdir"
@@ -45,6 +46,7 @@ setup_nvidia-nvrc() {
setup_nvidia-gpu-admin-tools() {
local TARGET="nvidia-gpu-admin-tools"
local TARGET_VERSION="v2024.12.06"
local TARGET_GIT="https://github.com/NVIDIA/gpu-admin-tools"
local TARGET_BUILD_DIR="${BUILD_DIR}/${TARGET}/builddir"
local TARGET_DEST_DIR="${BUILD_DIR}/${TARGET}/destdir"
@@ -72,6 +74,7 @@ setup_nvidia-gpu-admin-tools() {
setup_nvidia-dcgm-exporter() {
local TARGET="nvidia-dcgm-exporter"
local TARGET_VERSION="3.3.9-3.6.1"
local TARGET_BUILD_DIR="${BUILD_DIR}/${TARGET}/builddir"
local TARGET_DEST_DIR="${BUILD_DIR}/${TARGET}/destdir"
local TARBALL="${BUILD_DIR}/kata-static-${TARGET}.tar.zst"
@@ -85,7 +88,7 @@ setup_nvidia-dcgm-exporter() {
local dex="dcgm-exporter"
rm -rf "${dex}"
git clone https://github.com/NVIDIA/${dex}
git clone --branch "${TARGET_VERSION}" https://github.com/NVIDIA/${dex}
make -C ${dex} binary
mkdir -p ../destdir/bin
@@ -151,14 +154,8 @@ setup_nvidia_gpu_rootfs_stage_one() {
mount --make-rslave ./dev
mount -t proc /proc ./proc
local driver_version="latest"
if echo "$NVIDIA_GPU_STACK" | grep -q '\<latest\>'; then
driver_version="latest"
elif echo "$NVIDIA_GPU_STACK" | grep -q '\<lts\>'; then
driver_version="lts"
fi
chroot . /bin/bash -c "/nvidia_chroot.sh $(uname -r) ${run_file_name} ${run_fm_file_name} ${ARCH} ${driver_version}"
chroot . /bin/bash -c "/nvidia_chroot.sh $(uname -r) ${run_file_name} \
${run_fm_file_name} ${ARCH} ${NVIDIA_GPU_STACK}"
umount -R ./dev
umount ./proc