From 9917cecf151c4759deff171008de6ac6b93867ea Mon Sep 17 00:00:00 2001
From: Nikolaos Karaolidis <nick@karaolidis.com>
Date: Fri, 31 Jan 2025 09:55:01 +0000
Subject: [PATCH] Add nvidia-container-toolkit

Signed-off-by: Nikolaos Karaolidis <nick@karaolidis.com>
---
 hosts/eirene/hardware/default.nix | 15 +++++++++++++--
 1 file changed, 13 insertions(+), 2 deletions(-)

diff --git a/hosts/eirene/hardware/default.nix b/hosts/eirene/hardware/default.nix
index ae58d1b..889b43b 100644
--- a/hosts/eirene/hardware/default.nix
+++ b/hosts/eirene/hardware/default.nix
@@ -1,4 +1,9 @@
-{ pkgs, lib, ... }:
+{
+  config,
+  pkgs,
+  lib,
+  ...
+}:
 {
   imports = [
     ./display.nix
@@ -15,9 +20,9 @@
     };
 
     nvidia = {
+      open = false;
       modesetting.enable = true;
       powerManagement.enable = true;
-      open = false;
 
       prime = {
         offload = {
@@ -38,6 +43,12 @@
         rocmPackages.clr.icd
       ];
     };
+
+    nvidia-container-toolkit.enable = (
+      config.virtualisation.containerd.enable
+      || config.virtualisation.docker.enable
+      || config.virtualisation.podman.enable
+    );
   };
 
   boot = {