diff --git a/nix/impure-shell.nix b/nix/impure-shell.nix
index b0987e70589..9df4b111995 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