U yh@s4UddlZddlZddlZddlmZddlmZddlZddlm Z m Z m Z zddl m Z Wnek rvdZ YnXdaeeed<e dk rddZnd dZd d ZejejejhZd d ZddZdddZedddZGddde Zeejeeeje<eed<eed<eed<eed<eed<dS) N)contextmanager)Optional)__allow_nonbracketed_mutation ContextProp PropModule)_cudnn__cudnn_versionc stdkrtat}t}|\}}}|\}}}||krBd}n |dksPtjsZ||k}n||k}|stjdddkr|dSd|d|d }d tjkrtjd d t fd d dDrt |d|qt |dnt |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|]}|kVqdSN).0Z substringZld_library_pathrO/var/www/html/venv/lib/python3.8/site-packages/torch/backends/cudnn/__init__.py 9sz_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_inits:      rcCsdS)NFrrrrrrNscCsts dStS)zReturn the version of cuDNN.N)rrrrrrversionRsrcCstjjS)z9Return a bool indicating if CUDNN is currently available.)torch_CZ _has_cudnnrrrr is_available`srcCsltjsdS|jjdks$|jtkr(dStss( r>r8r;r9r<r:)NNNNN)FFr7FT) rr)r& contextlibrtypingrrZtorch.backendsrrrZtorch._Cr ImportErrorrint__annotations__rrZhalffloatdoubler%rr+r6r=r>modulesrFboolrrrrsT    4