fixes and improvements
This commit is contained in:
@@ -0,0 +1,23 @@
|
||||
{ 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";
|
||||
};
|
||||
}
|
||||
Reference in New Issue
Block a user