Updated docker-nvidia configuration

2024-08-15 08:48:03 -03:00
parent 41b7f85773
commit 5fc32bd6b9
3 changed files with 8 additions and 9 deletions
--- a/modules/containers-nvidia.nix
+++ b/modules/containers-nvidia.nix
@@ -1,7 +0,0 @@
-{...}: {
-  virtualisation.docker = {
-    enable = true;
-    storageDriver = "btrfs";
-    enableNvidia = true;
-  };
-}
--- a/modules/containers.nix
+++ b/modules/containers.nix
@@ -1,5 +1,11 @@
-{...}: {
+{
+  lib,
+  config,
+  ...
+}: {
  virtualisation.docker = {
    enable = true;
  };
+
+  hardware.nvidia-container-toolkit.enable = lib.mkIf config.hardware.nvidia.modesetting.enable true;
 }