9j UddlZddlZddlZddlmZddlmZddlZddlm Z m Z m Z m Z m Z mZddlmZ ddlmZdaedzed<ed Znd Zd Zej2ej4ej6hZd Zd Z ddZe ddZ GddeZ!e!ejDe#e#ejDe#<e$ed<e$ed<e$ed<e$ed<e%ed<eed<y#e$rdZYwxYw)N)contextmanager)Optional)__allow_nonbracketed_mutation_FP32Precision_get_fp32_precision_getter_set_fp32_precision_setter ContextProp PropModule)rnn)_cudnn__cudnn_versionc ttjatj}tj}|\}}}|\}}}||k7rd}n |dkstj s||k(}n||k\}|st jjdddk(ryd|d|d }d t jvrRt jjd d  t fd d Drt|d|t|dt|tjjrtjjs|t!t#tjj%D cgc]!} tjj'| #c} } tdk\r| dkrtdtdycc} w)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_PATHc3&K|]}|v ywN).0 substringld_library_paths ]/media/conek/DATA/Code/OCR/venv/lib/python3.12/site-packages/torch/backends/cudnn/__init__.py z_init..Gs9B _4s)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.ic)rzcuDNN version z is not compatible with devices with SM < 7.5. Please install a version of PyTorch with a compatible cuDNN version. https://github.com/pytorch/pytorch/blob/main/RELEASE.md#release-compatibility-matrix)rr getVersionIntgetRuntimeVersiongetCompileVersionis_cudaosenvirongetany RuntimeErrortorchr is_availableversionhipminrange device_countget_device_capability) runtime_versioncompile_version runtime_major runtime_minor_ compile_major compile_minorcudnn_compatiblebase_error_msgimin_ccrs @r_initr=$s  "$224O$668O$668O.= +M=!.= +M=! -#( "&..#0M#A #0M#A #::>>"JCPTWW55D4EF11@0AB34% 2&(jjnn5F&KOFW+-.VVeUfh +-.DE '~66zz&&(1B1B"'uzz'>'>'@!A 88; #e+&((9:op s=&G cy)NFrrrr=r=jsr?c$tsytS)zReturn the version of cuDNN.N)r=rrr?rr,r,ns 7 r?c6tjjS)z9Return a bool indicating if CUDNN is currently available.)r*_C _has_cudnnrr?rr+r+|s 88  r?ctjjsy|jjdk7s|j t vrytstjddytsItjdjddd jtjd  dyy ) NFrzPyTorch was compiled without cuDNN/MIOpen support. To use cuDNN/MIOpen, rebuild PyTorch making sure the library is visible to the build system.) stacklevelz4cuDNN/MIOpen library not found. Check your {libpath}DYLD_LIBRARY_PATHPATH)darwinwin32r)libpathT)r*rB_get_cudnn_enableddevicetypedtypeCUDNN_TENSOR_DTYPESr+warningswarnr=formatr'sysplatform)tensors r is_acceptablerWs 88 & & ( }}V#v||;N'N >  N  7 B I I#6HLLLL"3 J     r?c &tjjtjjt sdntjj tjj tjjtjjddf}|tjj||tjj||)t rtjj||tjj||tjj||!tjjdd||S)Nrall)r*rBrL_get_cudnn_benchmarkr+_cuda_get_cudnn_benchmark_limit_get_cudnn_deterministic_get_cudnn_allow_tf32r_set_cudnn_enabled_set_cudnn_benchmark_cuda_set_cudnn_benchmark_limit_set_cudnn_deterministic_set_cudnn_allow_tf32r)_enabled _benchmark_benchmark_limit_deterministic _allow_tf32_fp32_precision orig_flagss r set_flagsrjs ##% %%' N(P(P(R ))+ &&( ++FE: J ##H- %%j1#  001AB! )).9 &&{3" ++FE?K r?c #Kt5t||||||}ddd dt5tdddy#1swY*xYw#1swYyxYw#t5tdddw#1swYwxYwxYwwr)rrj)enabled benchmarkbenchmark_limit deterministic allow_tf32fp32_precisionris rflagsrrs ' (          # + , # z " # #   # # * , # z " # # #s_ B AB A B  A B A B AB  B* A<3 B<B BB cHeZdZeej j ej jZeej jej jZ eej jej jZdZer1eej j"ej j$Zeej j&ej j(ZeddZeeddeddZy) CudnnModuleNrconvrY)__name__ __module__ __qualname__r r*rBrLr^rlr\rarorZr_rmrnr+r[r`r]rbrprrurrrqrr?rrtrts%((55uxx7R7RSG ))588+L+LM %%uxx'D'DIO~% HH 4 4 HH 4 4  &&(F(FJ && )D "651"651Nr?rtrlrormrprqrn)NNNNNnone)FF FTry)&r%rTrQ contextlibrtypingrr*torch.backendsrrrrr r rr torch._Cr ImportErrorrint__annotations__r=r,halffloatdoublerPr+rWrjrrrtmodulesrvboolstrrr?rrs$ % #t" BL JJ KK LL 4 >  ##<*4$CKK$98D H    G FsC C*)C*