diff options
author | Averell Dalton <averell+nixpkgs@rxd4.com> | 2019-02-27 09:44:37 +0100 |
---|---|---|
committer | Averell Dalton <averell+nixpkgs@rxd4.com> | 2019-02-27 09:56:03 +0100 |
commit | 7f7209ef9ad16863d3a4c4e83a375e1795553bb7 (patch) | |
tree | 753e52a397e53579b2409a12cfc0d3de4cc4b454 /nixos/modules/virtualisation | |
parent | d6c2fe76ed75449865ccfd7878ab4a8b559d20ef (diff) | |
download | nixlib-7f7209ef9ad16863d3a4c4e83a375e1795553bb7.tar nixlib-7f7209ef9ad16863d3a4c4e83a375e1795553bb7.tar.gz nixlib-7f7209ef9ad16863d3a4c4e83a375e1795553bb7.tar.bz2 nixlib-7f7209ef9ad16863d3a4c4e83a375e1795553bb7.tar.lz nixlib-7f7209ef9ad16863d3a4c4e83a375e1795553bb7.tar.xz nixlib-7f7209ef9ad16863d3a4c4e83a375e1795553bb7.tar.zst nixlib-7f7209ef9ad16863d3a4c4e83a375e1795553bb7.zip |
nixos/docker: add enableNvidia option
Diffstat (limited to 'nixos/modules/virtualisation')
-rw-r--r-- | nixos/modules/virtualisation/docker.nix | 25 |
1 files changed, 22 insertions, 3 deletions
diff --git a/nixos/modules/virtualisation/docker.nix b/nixos/modules/virtualisation/docker.nix index a1a32c1c59a1..4ee84c5268e6 100644 --- a/nixos/modules/virtualisation/docker.nix +++ b/nixos/modules/virtualisation/docker.nix @@ -52,6 +52,15 @@ in ''; }; + enableNvidia = + mkOption { + type = types.bool; + default = false; + description = '' + Enable nvidia-docker wrapper, supporting NVIDIA GPUs inside docker containers. + ''; + }; + liveRestore = mkOption { type = types.bool; @@ -140,7 +149,8 @@ in ###### implementation config = mkIf cfg.enable (mkMerge [{ - environment.systemPackages = [ cfg.package ]; + environment.systemPackages = [ cfg.package ] + ++ optional cfg.enableNvidia pkgs.nvidia-docker; users.groups.docker.gid = config.ids.gids.docker; systemd.packages = [ cfg.package ]; @@ -157,6 +167,7 @@ in --log-driver=${cfg.logDriver} \ ${optionalString (cfg.storageDriver != null) "--storage-driver=${cfg.storageDriver}"} \ ${optionalString cfg.liveRestore "--live-restore" } \ + ${optionalString cfg.enableNvidia "--add-runtime nvidia=${pkgs.nvidia-docker}/bin/nvidia-container-runtime" } \ ${cfg.extraOptions} '']; ExecReload=[ @@ -165,7 +176,8 @@ in ]; }; - path = [ pkgs.kmod ] ++ (optional (cfg.storageDriver == "zfs") pkgs.zfs); + path = [ pkgs.kmod ] ++ optional (cfg.storageDriver == "zfs") pkgs.zfs + ++ optional cfg.enableNvidia pkgs.nvidia-docker; }; systemd.sockets.docker = { @@ -179,7 +191,6 @@ in }; }; - systemd.services.docker-prune = { description = "Prune docker resources"; @@ -194,7 +205,15 @@ in startAt = optional cfg.autoPrune.enable cfg.autoPrune.dates; }; + + assertions = [ + { assertion = cfg.enableNvidia -> config.hardware.opengl.driSupport32Bit or false; + message = "Option enableNvidia requires 32bit support libraries"; + }]; } + (mkIf cfg.enableNvidia { + environment.etc."nvidia-container-runtime/config.toml".source = "${pkgs.nvidia-docker}/etc/config.toml"; + }) ]); imports = [ |