From 4aca2038465009f23cf897c8949b39f0a241c200 Mon Sep 17 00:00:00 2001
From: Aargh Rai <aargh.rai+git@gmail.com>
Date: Thu, 3 Jul 2025 12:23:02 +0530
Subject: modulaized

---
 nvidia.nix | 26 ++++++++++++++++++++++++++
 1 file changed, 26 insertions(+)
 create mode 100644 nvidia.nix

(limited to 'nvidia.nix')

diff --git a/nvidia.nix b/nvidia.nix
new file mode 100644
index 0000000..74dd63f
--- /dev/null
+++ b/nvidia.nix
@@ -0,0 +1,26 @@
+{ config, pkgs, ... }:
+let unstable = import <nixos-unstable> { config.allowUnfree = true; };
+in {
+  hardware.graphics.enable = true;
+  hardware.nvidia = {
+    modesetting.enable = true;
+    package = config.boot.kernelPackages.nvidiaPackages.stable;
+    nvidiaSettings = true;
+    open = true;
+  };
+  boot.kernelParams = [ "acpi_backlight=none" ];
+  boot.initrd.kernelModules = [ "nvidia" ];
+  services.xserver.videoDrivers = [ "nvidia" ];
+  nixpkgs.config.cudaSupport = true;
+
+  environment.systemPackages = with pkgs; [
+    linuxPackages.nvidia_x11
+    unstable.cudaPackages.cuda_cudart
+    unstable.cudaPackages.cudatoolkit
+  ];
+  environment.sessionVariables = {
+    CUDA_HOME = "${pkgs.cudatoolkit}";
+    CUDA_PATH = "${pkgs.cudatoolkit}";
+    LD_LIBRARY_PATH = "${pkgs.linuxPackages.nvidia_x11}/lib";
+  };
+}
-- 
cgit v1.2.3