diff --git a/flake.nix b/flake.nix index 8bba91b..10e043f 100644 --- a/flake.nix +++ b/flake.nix @@ -104,7 +104,7 @@ noximilien = mkConfig "noximilien" "x86_64-linux" (with nm; [ common-pc common-pc-ssd common-cpu-intel ]); bolle = mkConfig "bolle" "x86_64-linux" (with nm; [ common-pc common-pc-ssd common-cpu-intel ]); nord = mkConfig "nord" "x86_64-linux" (with nm; [ common-pc common-pc-ssd common-cpu-intel-cpu-only common-gpu-amd ]); - sopp = mkConfig "sopp" "x86_64-linux" (with nm; [ common-pc common-pc-ssd common-cpu-intel ]); # TODO: common-gpu-nvidia + sopp = mkConfig "sopp" "x86_64-linux" (with nm; [ common-pc common-pc-ssd common-cpu-intel-cpu-only common-gpu-nvidia-nonprime ]); }; homeConfigurations = forAllSystems (system: let diff --git a/hardware/cuda.nix b/hardware/cuda.nix new file mode 100644 index 0000000..281a40a --- /dev/null +++ b/hardware/cuda.nix @@ -0,0 +1,48 @@ +{ config, pkgs, lib, ... }: +{ + # assumes common-gpu-nvidia from nixos-hardware is also added + # TODO: should we move it from flake.nix to here? + + # https://nixos.wiki/wiki/Nvidia + + nixpkgs.config.cudaSupport = true; + nixpkgs.config.allowUnfreePredicate = pkg: lib.any (x: x) [ + (lib.hasInfix "nvidia" (lib.toLower (lib.getName pkg))) + (lib.hasInfix "cuda" (lib.toLower (lib.getName pkg))) + (lib.hasInfix "cudnn" (lib.toLower (lib.getName pkg))) + (lib.hasInfix "cublas" (lib.toLower (lib.getName pkg))) + ]; + + hardware.opengl.enable = true; + hardware.nvidia.package = config.boot.kernelPackages.nvidiaPackages.stable; + hardware.nvidia.modesetting.enable = true; + + # add this to the host in question: + #hardware.nvidia.prime = { + # offload.enable = true; + # # Bus IDs. You can find them using lspci, grepping for "3D" or "VGA" + # intelBusId = "PCI:0:2:0"; + # nvidiaBusId = "PCI:1:0:0"; + #}; + + environment.systemPackages = with pkgs; ([ + nvtop + + ] ++ lib.optional hardware.nvidia.prime.offload.enable [ + (writeShellScriptBin "prime-run" '' + export __NV_PRIME_RENDER_OFFLOAD=1 + export __NV_PRIME_RENDER_OFFLOAD_PROVIDER=NVIDIA-G0 + export __GLX_VENDOR_LIBRARY_NAME=nvidia + export __VK_LAYER_NV_optimus=NVIDIA_only + exec "$@" + '') + # already added by nixos-hardware.common-gpu-nvidia + #(writeShellScriptBin "nvidia-offload" '' + # export __NV_PRIME_RENDER_OFFLOAD=1 + # export __NV_PRIME_RENDER_OFFLOAD_PROVIDER=NVIDIA-G0 + # export __GLX_VENDOR_LIBRARY_NAME=nvidia + # export __VK_LAYER_NV_optimus=NVIDIA_only + # exec "$@" + #'') + ]); +} diff --git a/hardware/rocm.nix b/hardware/rocm.nix index d5e523b..c24d805 100644 --- a/hardware/rocm.nix +++ b/hardware/rocm.nix @@ -5,6 +5,7 @@ # https://libreddit.noximilien.pbsds.net/r/archlinux/comments/nih9c9/amdgpu_vs_modesetting_in_current_510_kernels_xorg/ + hardware.opengl.enable = true; services.xserver.videoDrivers = [ "amdgpu" ]; # use gpu, adaptive sync and and hardware page flipping #services.xserver.videoDrivers = [ "modesetting" ]; # integrated (non-accelerated) framebuffer, KMS diff --git a/hosts/sopp/default.nix b/hosts/sopp/default.nix index 9f9410b..5aebe30 100644 --- a/hosts/sopp/default.nix +++ b/hosts/sopp/default.nix @@ -14,7 +14,7 @@ imports = [ ./hardware-configuration.nix - #../../hardware/rocm.nix # TODO: CUDA + ../../hardware/cuda.nix ../../hardware/opengl-intel.nix ../../users # home-manager