Add cuDNN

huggingface · Oct 22, 2024 · 6d60aa9 · 6d60aa9
1 parent 7aa90c5
commit 6d60aa9
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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