24 lines
621 B
Nix
24 lines
621 B
Nix
{ config, pkgs, ... }:
|
|
{
|
|
nix.settings = {
|
|
substituters = [
|
|
"https://nix-community.cachix.org"
|
|
];
|
|
trusted-public-keys = [
|
|
"nix-community.cachix.org-1:mB9FSh9qf2dCimDSUo8Zy7bkq5CX+/rkCWyvRCYg3Fs="
|
|
];
|
|
};
|
|
environment.systemPackages = with pkgs; [
|
|
cudaPackages.cudatoolkit
|
|
cudaPackages.cudnn
|
|
];
|
|
hardware.nvidia-container-toolkit.enable = true;
|
|
environment.variables = {
|
|
CUDA_HOME = "${pkgs.cudaPackages.cudatoolkit}";
|
|
CUDNN_PATH = "${pkgs.cudaPackages.cudnn}";
|
|
};
|
|
systemd.services.docker.serviceConfig = {
|
|
Environment = "DOCKER_CLI_EXPERIMENTAL=enabled";
|
|
};
|
|
}
|