16 lines
491 B
Nix
16 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" ];
|
||
|
|
};
|
||
|
|
};
|
||
|
|
}
|