diff --git a/Dockerfile b/Dockerfile
index 2a89d10..f5d40cd 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,6 +1,6 @@
 FROM quay.io/almalinuxorg/atomic-desktop-kde:10
 
-ARG KERNEL=kernel
+ARG KERNEL=kernel-ml
 ENV KERNEL=${KERNEL}
 
 RUN dnf upgrade -y
@@ -9,6 +9,7 @@ COPY bin/set_next_version.sh /tmp
 RUN /tmp/set_next_version.sh
 
 COPY repo/*.repo /etc/yum.repos.d/
+RUN dnf config-manager --add-repo=https://negativo17.org/repos/epel-nvidia.repo -y
 
 RUN dnf install --nogpgcheck -y https://mirrors.rpmfusion.org/free/el/rpmfusion-free-release-$(rpm -E %rhel).noarch.rpm https://mirrors.rpmfusion.org/nonfree/el/rpmfusion-nonfree-release-$(rpm -E %rhel).noarch.rpm
 
@@ -41,10 +42,18 @@ RUN dnf install -y https://github.com/TheAssassin/AppImageLauncher/releases/down
 # Enable CachyOS addons EL10 fork repo
 RUN dnf copr enable andersrh/kernel-cachyos-addons-el10 -y
 
-RUN dnf install -y ${KERNEL}-devel ${KERNEL}-headers
+RUN rpm --import https://www.elrepo.org/RPM-GPG-KEY-v2-elrepo.org
+RUN dnf install https://www.elrepo.org/elrepo-release-10.el10.elrepo.noarch.rpm -y
 
-RUN dnf install almalinux-release-nvidia-driver -y
-RUN dnf install nvidia-open-kmod nvidia-driver nvidia-driver-cuda -y
+RUN dnf install --enablerepo=elrepo-kernel -y ${KERNEL}
+
+RUN rpm -e --nodeps kernel kernel-core kernel-modules kernel-modules-core kernel-modules-extra
+
+RUN dnf install --enablerepo=elrepo-kernel -y ${KERNEL}-devel
+
+# Install Negativo17 Nvidia driver
+RUN dnf install -y dkms-nvidia nvidia-driver nvidia-persistenced opencl-filesystem libva-nvidia-driver
+RUN dkms install nvidia/$(ls /usr/src/ | grep nvidia- | cut -d- -f2-) -k $(rpm -q --queryformat "%{VERSION}-%{RELEASE}.%{ARCH}\n" ${KERNEL})
 
 RUN dnf install -y waydroid scx-scheds