From bc6d0fb4f6598c0c2459a18e446da11e35988eff Mon Sep 17 00:00:00 2001
From: GRMrGecko <grmrgecko@gmail.com>
Date: Tue, 18 Mar 2025 13:43:25 -0500
Subject: [PATCH] Fix docker nvidia.

---
 modules/nixos/common.nix               |  6 ++++++
 modules/nixos/video-drivers/nvidia.nix | 12 ++++++++++++
 2 files changed, 18 insertions(+)

diff --git a/modules/nixos/common.nix b/modules/nixos/common.nix
index 6450975..6d097ac 100644
--- a/modules/nixos/common.nix
+++ b/modules/nixos/common.nix
@@ -96,6 +96,12 @@
     '';
   };
 
+  # Added paths.
+  environment.etc."environment.d/00-custom-path.conf".text = ''
+    PATH="$PATH:/usr/local/bin:/usr/local/sbin:/usr/local/ubin"
+  '';
+
+
   # Nix Package Auto Cleanup
   nix = {
     settings.auto-optimise-store = true;
diff --git a/modules/nixos/video-drivers/nvidia.nix b/modules/nixos/video-drivers/nvidia.nix
index b452776..108a68f 100644
--- a/modules/nixos/video-drivers/nvidia.nix
+++ b/modules/nixos/video-drivers/nvidia.nix
@@ -4,9 +4,21 @@
   # Display drivers.
   hardware.graphics = {
     enable = true;
+    enable32Bit = true;
   };
   services.xserver.videoDrivers = [ "nvidia" ];
 
+  # Docker
+  hardware.nvidia-container-toolkit.enable = true;
+
+  environment.systemPackages = with pkgs; [
+    cudatoolkit
+    libGLU
+    libGL
+    libnvidia-container
+    nvidia-container-toolkit
+  ];
+
   hardware.nvidia = {
 
     # Modesetting is required.