{ pkgs, config, ... }:
{
  virtualisation.docker = {
    enable = true;
    storageDriver = "overlay2";
    enableOnBoot = true;
  };

  environment.systemPackages = with pkgs; [
    docker
    docker-compose
    docker-buildx
    nvidia-container-toolkit
  ];

  hardware.nvidia-container-toolkit.enable = true;
}