o I h@sFUddlZddlZddlZddlmZddlmZddlZddlm Z m Z m Z zddl m Z Wn ey:dZ Ynwdaeeed<e durLddZnd dZd d ZejejejhZd d ZddZ     dddZe     dddZGddde Zeejeeeje<eed<eed<eed<eed<eed<dS) N)contextmanager)Optional)__allow_nonbracketed_mutation ContextProp PropModule)_cudnn__cudnn_versionc stdurrtat}t}|\}}}|\}}}||kr!d}n|dks(tjs-||k}n||k}|srtjdddkr>dSd|d|d }d tjvrntjd d t fd d dDrgt |d|t |dt |dS)NF&PYTORCH_SKIP_CUDNN_COMPATIBILITY_CHECK01Tz=cuDNN version incompatibility: PyTorch was compiled against z but found runtime version z. PyTorch already comes bundled with cuDNN. One option to resolving this error is to ensure PyTorch can find the bundled cuDNN. LD_LIBRARY_PATHc3s|]}|vVqdSN).0 substringld_library_pathrQ/var/www/vscode/kcb/lib/python3.10/site-packages/torch/backends/cudnn/__init__.py :s z_init..)cudacudnnzLooks like your LD_LIBRARY_PATH contains incompatible version of cudnn. Please either remove it from the path or install cudnn zHone possibility is that there is a conflicting cuDNN in LD_LIBRARY_PATH.) rr getVersionIntgetRuntimeVersiongetCompileVersionis_cudaosenvirongetany RuntimeError) runtime_versioncompile_version runtime_major runtime_minor_ compile_major compile_minorcudnn_compatiblebase_error_msgrrr_initsF     r+cCsdS)NFrrrrrr+OscCstsdStS)zReturn the version of cuDNN.N)r+rrrrrversionSsr,cCstjjS)z9Return a bool indicating if CUDNN is currently available.)torch_C _has_cudnnrrrr is_availableasr0cCsltjsdS|jjdks|jtvrdStst ddSt s4t dj ddd t jdd dSd S) NFrzPyTorch was compiled without cuDNN/MIOpen support. To use cuDNN/MIOpen, rebuild PyTorch making sure the library is visible to the build system.z4cuDNN/MIOpen library not found. Check your {libpath}DYLD_LIBRARY_PATHPATH)darwinwin32r )libpathT)r-r._get_cudnn_enableddevicetypedtypeCUDNN_TENSOR_DTYPESr0warningswarnr+formatrsysplatform)tensorrrr is_acceptablefs&  rAcCstjtjts dntjtjtjf}|dur%tj||dur/tj ||durr; contextlibrtypingrr-torch.backendsrrrtorch._Cr ImportErrorrint__annotations__r+r,halffloatdoubler:r0rArQrXrYmodulesraboolrrrrsV      4