U Kc5@sUddlZddlZddlZddlZddlmZddlmZmZm Z zddl m Z Wne k rjdZ YnXda e dk rddZnddZdd ZejejejhZd d Zd d ZdddZedddZGdddeZeejeeeje<eed<eed<eed<eed<eed<dS)N)contextmanager) ContextProp PropModule__allow_nonbracketed_mutation)_cudnnc stdkrtat}t}|\}}}|\}}}||krBd}n |dksPtjsZ||k}n||k}|sd|d|d}dtjkrtjddt fdd d Drt |d |qt |d nt |d S)NFz=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_pathr A/tmp/pip-unpacked-wheel-gikjz4vx/torch/backends/cudnn/__init__.py 1sz_init..)cudaZcudnnz}Looks like your LD_LIBRARY_PATH contains incompatible version of cudnnPlease either remove it from the path or install cudnn zHone possibility is that there is a conflicting cuDNN in LD_LIBRARY_PATH.T) __cudnn_versionrZ getVersionIntZgetRuntimeVersionZgetCompileVersionZis_cudaosenvirongetany RuntimeError) Zruntime_versionZcompile_versionZ runtime_majorZ runtime_minor_Z compile_majorZ compile_minorZcudnn_compatibleZbase_error_msgr r r_inits(    rcCsdS)NFr r r r rr>scCsts dStS)zReturns the version of cuDNNN)rrr r r rversionBsrcCstjjS)z:Returns a bool indicating if CUDNN is currently available.)torch_CZ has_cudnnr r r r is_availablePsrcCsltjsdS|jjdks$|jtkr(dStss>   (