Function _preload_cuda_deps

python/pycolmap/__init__.py:36–45

Preloads CUDA dependencies from pip packages on Linux.

()

def _preload_cuda_deps():
    """Preloads CUDA dependencies from pip packages on Linux."""
    if platform.system() != "Linux":
        return
    cuda_libs = {
        "nvidia.cuda_runtime": "libcudart.so.*[0-9]",
        "nvidia.curand": "libcurand.so.*[0-9]",
    }
    for module_name, lib_glob in cuda_libs.items():
        _preload_cuda_lib(module_name, lib_glob)


_preload_cuda_deps()