o i] @s0ddlZddlZddlZdaeZddZdS)NFcCstrdSttr WddSdaWdn1swYtjdr+dSddlm}|j}|d}|\}}||j j krKt d||d}||krgt j d|d |d |d td d dSdS)a)Warn if the CUDA driver major version is older than cuda-bindings compile-time version. This function compares the CUDA major version that cuda-bindings was compiled against with the CUDA major version supported by the installed driver. If the compile-time major version is greater than the driver's major version, a warning is issued. The check runs only once per process. Subsequent calls are no-ops. The warning can be suppressed by setting the environment variable ``CUDA_PYTHON_DISABLE_MAJOR_VERSION_WARNING=1``. NT)CUDA_PYTHON_DISABLE_MAJOR_VERSION_WARNINGr)driveriz%Failed to query CUDA driver version: z/cuda-bindings was built for CUDA major version z1, but the NVIDIA driver only supports up to CUDA z. Some cuda-bindings features may not work correctly. Consider updating your NVIDIA driver, or using a cuda-bindings version built for CUDA zM. (Set CUDA_PYTHON_DISABLE_MAJOR_VERSION_WARNING=1 to suppress this warning.)) stacklevel)$_major_version_compatibility_checked_lockosenvironget cuda.bindingsr CUDA_VERSIONcuDriverGetVersionCUresult CUDA_SUCCESS RuntimeErrorwarningswarn UserWarning)rcompile_version compile_majorerrruntime_version runtime_majorrV/home/ubuntu/.local/lib/python3.10/site-packages/cuda/bindings/utils/_version_check.py#warn_if_cuda_major_version_mismatch s8     r)r threadingrrLockrrrrrrs