CI debug

Author: NicolasHug

packaging/pre_build_script.sh

@@ -8,11 +8,6 @@ set -ex
 # CMake helpers.
 conda install -y pybind11 -c conda-forge
 
-# Need to create a symlink for libnvcuvid.so.1 to libnvcuvid.so becuase that's
-# what Cmake is looking for and it doesn't exist by default. This is pretty
-# brittle.
-sudo ln -s /usr/lib64/libnvcuvid.so.1 /usr/lib64/libnvcuvid.so
-
 
 # Search for nvcuvid library in various locations for debugging CI build issues
 echo "[NVCUVID-SEARCH] === Searching for nvcuvid library ==="

src/torchcodec/_core/CMakeLists.txt

@@ -112,8 +112,26 @@ function(make_torchcodec_libraries
             ${CUDA_nppicc_LIBRARY}
         )
 
-        find_library(NVCUVID_LIBRARY NAMES nvcuvid REQUIRED)
-        message(STATUS "Found NVCUVID library: ${NVCUVID_LIBRARY}")
+        # Try the normal way first
+        find_library(NVCUVID_LIBRARY NAMES nvcuvid)
+        
+        # If not found, try with version suffix
+        if(NOT NVCUVID_LIBRARY)
+            find_library(NVCUVID_LIBRARY NAMES nvcuvid.1 PATHS /usr/lib64 /usr/lib)
+        endif()
+        
+        # Or specify the full path directly
+        if(NOT NVCUVID_LIBRARY)
+            set(NVCUVID_LIBRARY "/usr/lib64/libnvcuvid.so.1")
+        endif()
+        
+        if(NVCUVID_LIBRARY)
+            message(STATUS "Found NVCUVID: ${NVCUVID_LIBRARY}")
+        else()
+            message(FATAL_ERROR "Could not find NVCUVID library")
+        endif()
+        # find_library(NVCUVID_LIBRARY NAMES nvcuvid REQUIRED)
+        # message(STATUS "Found NVCUVID library: ${NVCUVID_LIBRARY}")
 
         # Add CUDA Driver library (needed for cuCtxGetCurrent, etc.)
         find_library(CUDA_DRIVER_LIBRARY NAMES cuda REQUIRED)