1 files changed, 6 insertions, 1 deletions
diff --git a/modules/nvidia.nix b/modules/nvidia.nix
index 7d6d0b0..0309c7a 100644
--- a/modules/nvidia.nix
+++ b/modules/nvidia.nix
@@ -3,6 +3,8 @@
 {
   services.xserver.videoDrivers = [ "nvidia" ];
 
+  hardware.graphics.enable32Bit = true;
+
   hardware.nvidia = {
     modesetting.enable = true;
     powerManagement.enable = true;
@@ -21,5 +23,8 @@
     };
   };
 
-  # environment.systemPackages = [ pkgs.cudatoolkit ];
+  environment.systemPackages = [
+    pkgs.cudatoolkit
+    pkgs.nvidia-container-toolkit
+  ];
 }