From 6d60aa97b9849debc211c4ba55b7c717e69e8094 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Dani=C3=ABl=20de=20Kok?= Date: Tue, 22 Oct 2024 08:02:48 +0000 Subject: [PATCH] Add cuDNN --- nix/impure-shell.nix | 1 + 1 file changed, 1 insertion(+) diff --git a/nix/impure-shell.nix b/nix/impure-shell.nix index b0987e70..9df4b111 100644 --- a/nix/impure-shell.nix +++ b/nix/impure-shell.nix @@ -68,6 +68,7 @@ mkShell { cuda_cccl cuda_cudart cuda_nvtx + cudnn libcublas libcusolver libcusparse