nix/modules/cuda.nix


1
2
3
4
5
6
7
8
9

{ config, lib, pkgs, ... }:
{
  environment.systemPackages = with pkgs; [
      cudatoolkit
      cudaPackages.cudnn
      cudaPackages.libcublas
      linuxPackages.nvidia_x11
  ];
}