forked from rarias/jungle
Rodrigo Arias Mallo
3269d763aa
The cudainfo program checks that we can initialize the CUDA RT library and communicate with the driver. It can be used as standalone program or built with cudainfo.gpuCheck so it is executed inside the build sandbox to see if it also works fine. It uses the autoAddDriverRunpath hook to inject in the runpath the location of the library directory for CUDA libraries. Reviewed-by: Aleix Boné <abonerib@bsc.es>
21 lines
854 B
Nix
21 lines
854 B
Nix
{ lib, config, pkgs, ... }:
|
|
{
|
|
# Configure Nvidia driver to use with CUDA
|
|
hardware.nvidia.package = config.boot.kernelPackages.nvidiaPackages.production;
|
|
hardware.nvidia.open = lib.mkDefault (builtins.abort "hardware.nvidia.open not set");
|
|
hardware.graphics.enable = true;
|
|
nixpkgs.config.nvidia.acceptLicense = true;
|
|
services.xserver.videoDrivers = [ "nvidia" ];
|
|
|
|
# enable support for derivations which require nvidia-gpu to be available
|
|
# > requiredSystemFeatures = [ "cuda" ];
|
|
programs.nix-required-mounts.enable = true;
|
|
programs.nix-required-mounts.presets.nvidia-gpu.enable = true;
|
|
# They forgot to add the symlink
|
|
programs.nix-required-mounts.allowedPatterns.nvidia-gpu.paths = [
|
|
config.systemd.tmpfiles.settings.graphics-driver."/run/opengl-driver"."L+".argument
|
|
];
|
|
|
|
environment.systemPackages = [ pkgs.cudainfo ];
|
|
}
|