o Zh@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)NFZ&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).0Z substringZld_library_pathrL/var/www/auris/lib/python3.10/site-packages/torch/backends/cudnn/__init__.py :s z_init..)cudaZcudnnzLooks 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.) rrZ getVersionIntZgetRuntimeVersionZgetCompileVersionZis_cudaosenvirongetany RuntimeError) Zruntime_versionZcompile_versionZ runtime_majorZ runtime_minor_Z compile_majorZ compile_minorZcudnn_compatibleZbase_error_msgrrr_initsF     rcCsdS)NFrrrrrrOscCstsdStS)zReturn the version of cuDNN.N)rrrrrrversionSsrcCstjjS)z9Return a bool indicating if CUDNN is currently available.)torch_CZ _has_cudnnrrrr is_availableasrcCsltjsdS|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}ZDYLD_LIBRARY_PATHPATH)darwinwin32r )ZlibpathT)rr_get_cudnn_enabledZdevicetypeZdtypeCUDNN_TENSOR_DTYPESrwarningswarnrformatrsysplatform)Ztensorrrr is_acceptablefs&  r+cCstjtjts dntjtjtjf}|dur%tj||dur/tj ||durs$     r>r8r;r9r<r:)NNNNN)FFr7FT) rr)r& contextlibrtypingrrZtorch.backendsrrrZtorch._Cr ImportErrorrint__annotations__rrZhalffloatdoubler%rr+r6r=r>modulesrFboolrrrrsV      4