feat: add cuda setup
This commit is contained in:
parent
4d0a35652c
commit
d5782ada85
4 changed files with 33 additions and 2 deletions
|
|
@ -24,6 +24,7 @@ in
|
|||
./hardware-configuration.nix
|
||||
./boot.nix
|
||||
../../modules
|
||||
../../modules/cuda.nix
|
||||
(import ../../modules/nvidia.nix {
|
||||
nvidia-mode = nvidia-mode;
|
||||
intel-bus-id = intel-bus-id;
|
||||
|
|
|
|||
21
system/modules/cuda.nix
Normal file
21
system/modules/cuda.nix
Normal file
|
|
@ -0,0 +1,21 @@
|
|||
# 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"
|
||||
];
|
||||
}
|
||||
|
|
@ -7,6 +7,9 @@
|
|||
{
|
||||
environment.systemPackages =
|
||||
(with pkgs; [
|
||||
# Binary cache platform
|
||||
cachix
|
||||
|
||||
# gtk theme
|
||||
gtk3
|
||||
adwaita-icon-theme
|
||||
|
|
@ -90,7 +93,7 @@
|
|||
# Thumbnail
|
||||
ffmpegthumbnailer
|
||||
])
|
||||
++ ([
|
||||
++ [
|
||||
inputs.ghostty.packages.${system}.default
|
||||
]);
|
||||
];
|
||||
}
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue