remove unnecessary cuda_runtime.h include, update cmake accordingly

Author: NicolasHug

src/torchcodec/_core/BetaCudaDeviceInterface.cpp

@@ -14,7 +14,7 @@
 #include "src/torchcodec/_core/FFMPEGCommon.h"
 #include "src/torchcodec/_core/NVDECCache.h"
 
-#include <cuda_runtime.h> // For cudaStreamSynchronize
+// #include <cuda_runtime.h> // For cudaStreamSynchronize
 #include "src/torchcodec/_core/nvcuvid_include/cuviddec.h"
 #include "src/torchcodec/_core/nvcuvid_include/nvcuvid.h"
 

src/torchcodec/_core/CMakeLists.txt

@@ -107,11 +107,6 @@ function(make_torchcodec_libraries
     )
 
     if(ENABLE_CUDA)
-        list(APPEND core_library_dependencies
-            ${CUDA_nppi_LIBRARY}
-            ${CUDA_nppicc_LIBRARY}
-        )
-
         # Try to find NVCUVID. Try the normal way first. This should work locally.
         find_library(NVCUVID_LIBRARY NAMES nvcuvid)
         # If not found, try with version suffix, or hardcoded path. Appears
@@ -129,11 +124,11 @@ function(make_torchcodec_libraries
             message(FATAL_ERROR "Could not find NVCUVID library")
         endif()
 
-        # Add CUDA Driver library (needed for cuCtxGetCurrent, etc.)
-        find_library(CUDA_DRIVER_LIBRARY NAMES cuda REQUIRED)
-        message(STATUS "Found CUDA Driver library: ${CUDA_DRIVER_LIBRARY}")
-
-        list(APPEND core_library_dependencies ${NVCUVID_LIBRARY} ${CUDA_DRIVER_LIBRARY})
+        list(APPEND core_library_dependencies
+            ${CUDA_nppi_LIBRARY}
+            ${CUDA_nppicc_LIBRARY}
+            ${NVCUVID_LIBRARY}
+        )
     endif()
 
     make_torchcodec_sublibrary(