Pin NPP version for CUDA 12 (#5480)

Signed-off-by: Kamil Tokarski <ktokarski@nvidia.com>
NVIDIA · May 24, 2024 · 1f4a763 · 1f4a763
1 parent 814d004
commit 1f4a763
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/qa/test_template_impl.sh b/qa/test_template_impl.sh
@@ -156,7 +156,8 @@ do
         version_ge "${CUDA_VERSION}" "110" && \
           install_pip_pkg "pip install --upgrade nvidia-nvimgcodec-cu${DALI_CUDA_MAJOR_VERSION} -f /pip-packages" && \
           if [ "$(uname -m)" == "x86_64" ] && [ -z "${DO_NOT_INSTALL_CUDA_WHEEL}" ] && [ -z "${CONDA_PREFIX}" ]; then
-            install_pip_pkg "pip install --upgrade nvidia-npp-cu${DALI_CUDA_MAJOR_VERSION}    \
+            NPP_VERSION=$(if [[ $DALI_CUDA_MAJOR_VERSION == "12" ]]; then echo "==12.2.5.30"; else echo ""; fi)
+            install_pip_pkg "pip install --upgrade nvidia-npp-cu${DALI_CUDA_MAJOR_VERSION}${NPP_VERSION}    \
                                                    nvidia-nvjpeg-cu${DALI_CUDA_MAJOR_VERSION} \
                                                    nvidia-cufft-cu${DALI_CUDA_MAJOR_VERSION}  \
                                                    -f /pip-packages"