From 1f4a763f7204719672cc30701d4b14f8be7f4f33 Mon Sep 17 00:00:00 2001 From: Kamil Tokarski Date: Fri, 24 May 2024 14:20:29 +0200 Subject: [PATCH] Pin NPP version for CUDA 12 (#5480) Signed-off-by: Kamil Tokarski --- qa/test_template_impl.sh | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/qa/test_template_impl.sh b/qa/test_template_impl.sh index ef243380ba..547ab354fe 100755 --- 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"