flake: added cudaSupport for config. Enable cuda on Architect

This commit is contained in:
Giulio De Pasquale 2024-03-12 18:43:00 +00:00
parent d76b08a80d
commit 925502ccc4

View File

@ -15,9 +15,10 @@
sysDarwin = "aarch64-darwin";
sysLinuxAarch = "aarch64-linux";
wrapPkgsSystem = { system }:
wrapPkgsSystem = { system, cudaSupport ? false }:
let
config = {
inherit cudaSupport;
allowUnfree = true;
};
in
@ -53,8 +54,8 @@
});
pkgsLinuxX64 = wrapPkgsSystem { system = sysLinuxX64; };
utilsLinuxX64 = wrapUtils { system = sysLinuxX64; pkgs = pkgsLinuxX64; };
pkgsLinuxX64Cuda = wrapPkgsSystem { system = sysLinuxX64; cudaSupport = true; };
utilsLinuxX64Cuda = wrapUtils { system = sysLinuxX64; pkgs = pkgsLinuxX64Cuda; };
pkgsLinuxAarch = wrapPkgsSystem { system = sysLinuxAarch; };
utilsLinuxAarch = wrapUtils { system = sysLinuxAarch; pkgs = pkgsLinuxAarch; };
@ -64,14 +65,14 @@
in
{
nixosConfigurations = {
architect = utilsLinuxX64.host.mkHost {
architect = utilsLinuxX64Cuda.host.mkHost {
name = "architect";
users = [{
user = "giulio";
roles = [ ];
}];
};
gAluminum = utilsLinuxX64.host.mkHost {
gAluminum = utilsLinuxX64Cuda.host.mkHost {
name = "gAluminum";
users = [{
user = "giulio";
@ -90,11 +91,11 @@
name = "giulio";
roles = [ "ssh" ];
};
giulioX64 = utilsLinuxX64.user.mkHMUser {
giulioX64 = utilsLinuxX64Cuda.user.mkHMUser {
name = "giulio";
roles = [ "ssh" "go" ];
};
giulioX64NoSSH = utilsLinuxX64.user.mkHMUser {
giulioX64NoSSH = utilsLinuxX64Cuda.user.mkHMUser {
name = "giulio";
roles = [ "go" ];
};