nvImageCodec related fixes

Signed-off-by: Joaquin Anton Guirao <janton@nvidia.com>

dali/python/nvidia/dali/backend.py

@@ -117,22 +117,3 @@ def check_cuda_runtime():
                 "#pip-wheels-installation-linux "
                 "for the reference."
             )
-
-
-nvimgcodec_checked = False
-
-
-def check_nvimgcodec():
-    """
-    Checks the availability of nvImageCodec library, and prints an appropriate warning.
-    """
-    global nvimgcodec_checked
-    if not nvimgcodec_checked:
-        nvimgcodec_checked = True
-        if GetNvimgcodecVersion() == -1:
-            deprecation_warning(
-                "DALI's experimental image decoder functionality now requires NVIDIA nvImageCodec "
-                "library to run. You need to install it separately. "
-                "See https://developer.nvidia.com/nvimgcodec-downloads or simply do "
-                f"`pip install nvidia-nvimgcodec-cu{str(__cuda_version__)[:2]}`"
-            )

dali/python/nvidia/dali/pipeline.py

@@ -855,7 +855,6 @@ def _start_py_workers(self):
 
     def _init_pipeline_backend(self):
         device_id = self._device_id if self._device_id is not None else types.CPU_ONLY_DEVICE_ID
-        b.check_nvimgcodec()
         if device_id != types.CPU_ONLY_DEVICE_ID:
             b.check_cuda_runtime()
         self._pipe = b.Pipeline(

docker/Dockerfile

@@ -80,8 +80,8 @@ ARG BUILD_NVCOMP
 ENV BUILD_NVCOMP=${BUILD_NVCOMP}
 ARG BUILD_CVCUDA
 ENV BUILD_CVCUDA=${BUILD_CVCUDA}
-ARG BUILD_NVIMGCODEC
-ENV BUILD_NVIMGCODEC=${BUILD_NVIMGCODEC}
+ARG BUILD_NVIMAGECODEC
+ENV BUILD_NVIMAGECODEC=${BUILD_NVIMAGECODEC}
 ARG LINK_DRIVER
 ENV LINK_DRIVER=${LINK_DRIVER}
 ARG WITH_DYNAMIC_CUDA_TOOLKIT

docker/Dockerfile.build.aarch64-linux

@@ -140,7 +140,7 @@ CMD WERROR=ON           \
     BUILD_NVML=OFF      \
     VERBOSE_LOGS=OFF    \
     BUILD_CUFILE=OFF    \
-    BUILD_NVIMGCODEC=ON \
+    BUILD_NVIMAGECODEC=ON \
     TEST_BUNDLED_LIBS=NO \
     WITH_DYNAMIC_CUDA_TOOLKIT=ON \
     WITH_DYNAMIC_NVJPEG=ON \

docker/build.sh

@@ -173,7 +173,7 @@ if [ "$BUILD_INHOST" == "YES" ]; then
                                         BUILD_CFITSIO=${BUILD_CFITSIO}            \
                                         BUILD_CUFILE=${BUILD_CUFILE}              \
                                         BUILD_NVCOMP=${BUILD_NVCOMP}              \
-                                        BUILD_NVIMGCODEC=${BUILD_NVIMGCODEC}      \
+                                        BUILD_NVIMAGECODEC=${BUILD_NVIMAGECODEC}      \
                                         LINK_DRIVER=${LINK_DRIVER}                \
                                         WITH_DYNAMIC_CUDA_TOOLKIT=${WITH_DYNAMIC_CUDA_TOOLKIT} \
                                         WITH_DYNAMIC_NVJPEG=${WITH_DYNAMIC_NVJPEG:-ON} \

docker/build_helper.sh

@@ -31,6 +31,7 @@ export BUILD_BENCHMARK=${BUILD_BENCHMARK:-ON}
 export BUILD_NVTX=${BUILD_NVTX}
 export BUILD_PYTHON=${BUILD_PYTHON:-ON}
 export BUILD_LMDB=${BUILD_LMDB:-ON}
+export BUILD_NVIMAGECODEC=${BUILD_NVIMAGECODEC:-ON}
 export BUILD_JPEG_TURBO=${BUILD_JPEG_TURBO:-ON}
 export BUILD_OPENCV=${BUILD_OPENCV:-ON}
 export BUILD_PROTOBUF=${BUILD_PROTOBUF:-ON}
@@ -48,7 +49,6 @@ export BUILD_CFITSIO=${BUILD_CFITSIO:-ON}
 export BUILD_CVCUDA=${BUILD_CVCUDA:-ON}
 export BUILD_CUFILE=${BUILD_CUFILE:-OFF}
 export BUILD_NVCOMP=${BUILD_NVCOMP:-OFF}
-export BUILD_NVIMGCODEC=${BUILD_NVIMGCODEC:-OFF}
 export LINK_LIBCUDA=${LINK_LIBCUDA:-OFF}
 export WITH_DYNAMIC_CUDA_TOOLKIT=${WITH_DYNAMIC_CUDA_TOOLKIT:-OFF}
 export WITH_DYNAMIC_NVJPEG=${WITH_DYNAMIC_NVJPEG:-ON}
@@ -85,6 +85,7 @@ cmake ../ -DCMAKE_INSTALL_PREFIX=.                 \
       -DBUILD_NVTX=${BUILD_NVTX}                   \
       -DBUILD_PYTHON=${BUILD_PYTHON}               \
       -DBUILD_LMDB=${BUILD_LMDB}                   \
+      -DBUILD_NVIMAGECODEC=${BUILD_NVIMAGECODEC}   \
       -DBUILD_JPEG_TURBO=${BUILD_JPEG_TURBO}       \
       -DBUILD_OPENCV=${BUILD_OPENCV}               \
       -DBUILD_PROTOBUF=${BUILD_PROTOBUF}           \

docs/compilation.rst

@@ -338,7 +338,7 @@ Optional CMake Build Parameters
 -  ``BUILD_NVDEC`` - build with ``NVIDIA NVDEC`` support (default: ON)
 -  ``BUILD_NVML`` - build with ``NVIDIA Management Library`` (``NVML``) support (default: ON)
 -  ``BUILD_CUFILE`` - build with ``GPU Direct Storage`` support (default: ON)
--  ``BUILD_NVIMGCODEC`` - build with ``NVIDIA nvImageCodec library`` support (default: ON)
+-  ``BUILD_NVIMAGECODEC`` - build with ``NVIDIA nvImageCodec library`` support (default: ON)
 -  ``VERBOSE_LOGS`` - enables verbose loging in DALI. (default: OFF)
 -  ``WERROR`` - treat all build warnings as errors (default: OFF)
 -  ``BUILD_DALI_NODEPS`` - disables support for third party libraries that are normally expected to be available in the system