diff --git a/stacks/pinot-noir/modules_acc_cuda.yaml b/stacks/pinot-noir/modules_acc_cuda.yaml index dd76dbbef2f0a588e4198e7a96b6f0a11dfe8563..d098e12fa90540a012f5d36253acde831cbd85dd 100644 --- a/stacks/pinot-noir/modules_acc_cuda.yaml +++ b/stacks/pinot-noir/modules_acc_cuda.yaml @@ -21,3 +21,12 @@ modules: environment: set: NCCL_HOME: "{prefix}" + # needed as long as drivers are with 12.2 support + cuda@12.4: + environment: + prepend_path: + LD_LIBRARY_PATH: /ssoft/spack/external/cuda-compat/12.4/lib64 + cuda@12.5: + environment: + prepend_path: + LD_LIBRARY_PATH: /ssoft/spack/external/cuda-compat/12.6/lib64