21 lines
528 B
Nix
21 lines
528 B
Nix
# Here only set the environment for cuda shell env
|
|
# For futher, see flake: https://github.com/DACHXY/python-cuda-flake
|
|
{ lib, ... }:
|
|
{
|
|
nix = {
|
|
settings = {
|
|
substituters = [
|
|
"https://cuda-maintainers.cachix.org"
|
|
"https://cache.nixos.org/"
|
|
];
|
|
trusted-public-keys = [
|
|
"cuda-maintainers.cachix.org-1:0dq3bujKpuEPMCX6U4WylrUDZ9JyUG0VpVZa7CNfq5E="
|
|
];
|
|
};
|
|
};
|
|
nixpkgs.config.allowUnfreePredicate =
|
|
pkg:
|
|
builtins.elem (lib.getName pkg) [
|
|
"cuda-merged"
|
|
];
|
|
}
|