diff --git a/modules/nixos/hardware/nvidia.nix b/modules/nixos/hardware/nvidia.nix
index 5a26043..3104c01 100644
--- a/modules/nixos/hardware/nvidia.nix
+++ b/modules/nixos/hardware/nvidia.nix
@@ -11,6 +11,11 @@
       enable = true;
     };
 
+    nixpkgs.config = {
+      cudaSupport = true;
+      cudaCapabilities = [ "8.9" ]; # 4090
+    };
+
     # Load nvidia driver for Xorg and Wayland
     services.xserver.videoDrivers = [ "nvidia" ];