{ config, pkgs, ... }:
{
  nix.settings = {
    substituters = [
      "https://nix-community.cachix.org"
    ];
    trusted-public-keys = [
      "nix-community.cachix.org-1:mB9FSh9qf2dCimDSUo8Zy7bkq5CX+/rkCWyvRCYg3Fs="
    ];
  };
  environment.systemPackages = with pkgs; [
    cudaPackages.cudatoolkit
    cudaPackages.cudnn
  ];
  hardware.nvidia-container-toolkit.enable = true;
  environment.variables = {
    CUDA_HOME = "${pkgs.cudaPackages.cudatoolkit}";
    CUDNN_PATH = "${pkgs.cudaPackages.cudnn}";
  };
  systemd.services.docker.serviceConfig = {
    Environment = "DOCKER_CLI_EXPERIMENTAL=enabled";
  };
}