diff --git a/stacks/pinot-noir-gcc/definitions_cuda.yaml b/stacks/pinot-noir-gcc/definitions_cuda.yaml index 4710276c2be849f7ba905a7c66ce192ee3a206eb..bbb3a547e4038400c5e8f7063df89eb30eaf0425 100644 --- a/stacks/pinot-noir-gcc/definitions_cuda.yaml +++ b/stacks/pinot-noir-gcc/definitions_cuda.yaml @@ -2,6 +2,7 @@ definitions: - cuda_system_codes: - cuda - cudnn + - cudnn@9.2.1.18-12 - cuda_serial_codes: - nccl