nix/modules/cuda.nix


1
2
3
4
5
6
7
8
9

{ config, lib, pkgs, ... }:
{
  environment.systemPackages = (if config.monorepo.profiles.cuda.enable then with pkgs; [
	cudatoolkit
	cudaPackages.cudnn
	cudaPackages.libcublas
	linuxPackages.nvidia_x11
  ] else []);
}