nix/modules/nixos/services/containerised/nvsample/default.nix
Ameya Shenoy 4998e822a7 feat: init
Signed-off-by: Ameya Shenoy <shenoy.ameya@gmail.com>
2024-11-11 01:48:40 +05:30

15 lines
491 B
Nix

{ config, lib, ... }: {
options.snowflake.services.containerised.nvsample = {
enable = lib.mkEnableOption "enable nvsample";
};
config = lib.mkIf config.snowflake.services.containerised.nvsample.enable {
virtualisation.oci-containers.containers.nvsample = {
autoStart = true;
cmd = [ "-c" "while true; do nvidia-smi; sleep 1; done" ];
entrypoint = "sh";
image = "docker.io/ubuntu";
extraOptions = [ "--device=nvidia.com/gpu=all" ];
};
};
}