diff --git a/config/sources/families/uefi-x86.conf b/config/sources/families/uefi-x86.conf
index 36ca9caca..55923fd7b 100644
--- a/config/sources/families/uefi-x86.conf
+++ b/config/sources/families/uefi-x86.conf
@@ -1,4 +1,5 @@
 # Important: LINUXFAMILY and ARCH are defined _before_ including the common family include
+[[ "$BUILD_DESKTOP" == yes ]] && enable_extension "nvidia"
 export LINUXFAMILY="x86"
 export ARCH="amd64"
 source "${BASH_SOURCE%/*}/include/uefi_common.inc"
diff --git a/extensions/nvidia.sh b/extensions/nvidia.sh
new file mode 100644
index 000000000..2e3c55487
--- /dev/null
+++ b/extensions/nvidia.sh
@@ -0,0 +1,13 @@
+function pre_install_kernel_debs__build_nvidia_kernel_module() {
+
+	export INSTALL_HEADERS="yes"
+
+}
+
+function post_install_kernel_debs__build_nvidia_kernel_module() {
+
+	display_alert "Build kernel module" "${EXTENSION}" "info"
+	chroot "${SDCARD}" /bin/bash -c "apt -y -qq install nvidia-dkms-510 nvidia-driver-510 nvidia-settings nvidia-common" >>"$DEST"/"${LOG_SUBPATH}"/install.log 2>&1 || {
+		exit_with_error "${install_grub_cmdline} failed!"
+	}
+}