Skip to content

Commit d9d1073

Browse files
manuelh-devfidencio
authored andcommitted
gpu: Install packages for devkit
Introduce a new function to install additional packages into the devkit flavor. With modprobe, we avoid errors on pod startup related to loading nvidia kernel modules in the NVRC phase. Note, the production flavor gets modprobe from busybox, see its configuration file containing CONFIG_MODPROBE=y. Signed-off-by: Manuel Huber <manuelh@nvidia.com>
1 parent a786582 commit d9d1073

File tree

1 file changed

+13
-0
lines changed

1 file changed

+13
-0
lines changed

tools/osbuilder/rootfs-builder/nvidia/nvidia_chroot.sh

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -147,6 +147,18 @@ install_nvidia_dcgm() {
147147
datacenter-gpu-manager-exporter
148148
}
149149

150+
install_devkit_packages() {
151+
is_feature_enabled "devkit" || {
152+
echo "chroot: Skipping devkit packages installation"
153+
return
154+
}
155+
156+
echo "chroot: Install devkit packages"
157+
158+
eval "${APT_INSTALL}" kmod
159+
apt-mark hold kmod
160+
}
161+
150162
cleanup_rootfs() {
151163
echo "chroot: Cleanup NVIDIA GPU rootfs"
152164

@@ -174,4 +186,5 @@ install_userspace_components
174186
install_nvidia_fabricmanager
175187
install_nvidia_ctk
176188
install_nvidia_dcgm
189+
install_devkit_packages
177190
cleanup_rootfs

0 commit comments

Comments
 (0)