foo.nix

{ pkgs, config, ... }:
{
  nixpkgs.config.allowUnfree = true;
  nixpkgs.config.nvidia.acceptLicense = true;
  # enable the nvidia driver
  services.xserver.videoDrivers = [ "nvidia" ];
  hardware.opengl.enable = true;
  hardware.nvidia.datacenter.enable = true;
  hardware.nvidia.package = config.boot.kernelPackages.nvidiaPackages.dc_535;

  hardware.nvidia.open = true;

  virtualisation.docker.enable = true;
  hardware.nvidia-container-toolkit.enable = true;
  hardware.opengl.driSupport32Bit = true;

  # QEMU options
  #virtualisation.qemu.options = [];

  nixos-shell.mounts.extraMounts = {
    # override options for each mount
    "/llm-os" = {
      target = ../.;
      cache = "none";
    };

  };

  environment.systemPackages = [ config.boot.kernelPackages.nvidia_x11 ];
}