Added nvidia-dkms when installing propri driver to (#2241)

* Added nvidia-dkms when installing propri driver to fix #2233, #2214 and #2002

* Forgot class-name infront of NvidiaDKMS

1 files changed, 5 insertions, 1 deletions

diff --git a/archinstall/lib/hardware.py b/archinstall/lib/hardware.py
index 56d3bc7b..d9342a98 100644
--- a/archinstall/lib/hardware.py
+++ b/archinstall/lib/hardware.py
@@ -42,6 +42,7 @@ class GfxPackage(Enum):
 	LibvaMesaDriver = 'libva-mesa-driver'
 	Mesa = "mesa"
 	Nvidia = 'nvidia'
+	NvidiaDKMS = 'nvidia-dkms'
 	NvidiaOpen = 'nvidia-open'
 	VulkanIntel = 'vulkan-intel'
 	VulkanRadeon = 'vulkan-radeon'
@@ -108,7 +109,10 @@ class GfxDriver(Enum):
 					GfxPackage.LibvaMesaDriver
 				]
 			case GfxDriver.NvidiaProprietary:
-				return [GfxPackage.Nvidia]
+				return [
+					GfxPackage.Nvidia,
+					GfxPackage.NvidiaDKMS
+				]
 			case GfxDriver.VMOpenSource:
 				return [
 					GfxPackage.Mesa,