sgUddlZddlZddlZddlmZddlmZddlZddlm Z m Z m Z  ddl m Z daeeed<e dZndZd Zej(ej*ej,hZd Zd Z dd Ze dd ZGdde Zeej:eeej:e<eed<eed<eed<eed<eed<y#e$rdZ YwxYw)N)contextmanager)Optional)__allow_nonbracketed_mutation ContextProp PropModule)_cudnn__cudnn_versionc0 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|y)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 P/var/www/html/venv/lib/python3.12/site-packages/torch/backends/cudnn/__init__.py z_init..:s9B _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.) r r getVersionIntgetRuntimeVersiongetCompileVersionis_cudaosenvirongetany RuntimeError) runtime_versioncompile_version runtime_major runtime_minor_ compile_major compile_minorcudnn_compatiblebase_error_msgrs @r_initr-sU  "$224O$668O$668O.= +M=!.= +M=! -#( "&..#0M#A #0M#A #::>>"JCPTWW55D4EF11@0AB34% 2&(jjnn5F&KOFW+-.VVeUfh +-.DE '~66cy)NFrrr.rr-r-Osr.c$tsytS)zReturn the version of cuDNN.N)r-r rr.rversionr1Ss 7 r.c6tjjS)z9Return a bool indicating if CUDNN is currently available.)torch_C _has_cudnnrr.r is_availabler6as 88  r.c|tjjsy|jjdk7s|j t vrytstjdytsGtjdjdddjtjd yy ) 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)r3r4_get_cudnn_enableddevicetypedtypeCUDNN_TENSOR_DTYPESr6warningswarnr-formatr!sysplatform)tensors r is_acceptablerHfs 88 & & ( }}V#v||;N'N >  N  7 B I I#6HLLLL"3 J   r.ctjjtjjt sdntjj tjj tjjf}|tjj||tjj||)t rtjj||tjj||tjj||Sr) r3r4r=_get_cudnn_benchmarkr6_cuda_get_cudnn_benchmark_limit_get_cudnn_deterministic_get_cudnn_allow_tf32_set_cudnn_enabled_set_cudnn_benchmark_cuda_set_cudnn_benchmark_limit_set_cudnn_deterministic_set_cudnn_allow_tf32)_enabled _benchmark_benchmark_limit_deterministic _allow_tf32 orig_flagss r set_flagsrY}s ##% %%' N(P(P(R ))+ &&( J ##H- %%j1#  001AB! )).9 &&{3 r.c#Kt5t|||||}ddd dt5tdddy#1swY*xYw#1swYyxYw#t5tdddw#1swYwxYwxYwwr)rrY)enabled benchmarkbenchmark_limit deterministic allow_tf32rXs rflagsr`s ' (  Y   # + , # z " # #   # # * , # z " # # #s_ B AB A B  A B A B AB  B) A;2 B;B BB ceZdZfdZeej jej jZ eej jej jZ eej jej jZdZer1eej j$ej j&Zeej j(ej j*ZxZS) CudnnModulec&t|||yr)super__init__)selfmname __class__s rrezCudnnModule.__init__s D!r.N)__name__ __module__ __qualname__rerr3r4r=rNr[rLrQr^rJrOr\r]r6rKrPrMrRr_ __classcell__)ris@rrbrbs"%((55uxx7R7RSG ))588+L+LM %%uxx'D'DIO~% HH 4 4 HH 4 4  &&(F(FJr.rbr[r^r\r_r])NNNNN)FF FT) rrErB contextlibrtypingrr3torch.backendsrrrtorch._Cr ImportErrorr int__annotations__r-r1halffloatdoublerAr6rHrYr`rbmodulesrjboolrr.rr{s % QQ"&#% 0h JJ KK LL 0 6  ##0*0$CKK$98D H    E FsC CC