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 = {