Lj"UddlZddlZddlZddlmZddlmZddlZddlm Z m Z m Z m Z m Z mZddlmZ ddlmZn #e$rdZYnwxYwdaedzed<ed Znd Zd ZejejejhZd Zd Z ddZe ddZ GddeZ!e!ej"e#e#ej"e#<e$ed<e$ed<e$ed<e$ed<e%ed<eed<e%ed<dS) N)contextmanager)Optional)__allow_nonbracketed_mutation_FP32Precision_get_fp32_precision_getter_set_fp32_precision_setter ContextProp PropModule)rnn)_cudnn__cudnn_versionc ttjatj}tj}|\}}}|\}}}||krd}n|dks tjs||k}n||k}|st jdddkrdSd|d|d }d t jvrat jd d  t fd d Drt|d|t|dt|tj r|tjjskt!dt#tj D} tdkr| dkrtdtddS)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|]}|vV dSN).0 substringld_library_paths KC:\Projects\DSCons\.venv\Lib\site-packages\torch\backends\cudnn\__init__.py z_init..Gs99B _4)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.cLg|]!}tj|"Sr)torchrget_device_capability)ris r z_init..Zs8 88;;ric)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 RuntimeErrorr"r is_availableversionhipminrange device_count) runtime_versioncompile_version runtime_major runtime_minor_ compile_major compile_minorcudnn_compatiblebase_error_msgmin_ccrs @r_initr@$sR  "$244O$688O$688O.= +M=!.= +M=! --#(  ""&."#0M#A  #0M#A # 7:>>"JCPPTWWW445D441@444% 22&(jnn5F&K&KOFW +-hhVehh +-EEE '~666z&&(( 1B !&uz'>'>'@'@!A!A #e++&pppp trcdS)NFrrrrr@r@jsurc0tsdStS)zReturn the version of cuDNN.N)r@rrrrr1r1ns 77t rc$tjjS)z9Return a bool indicating if CUDNN is currently available.)r"_C _has_cudnnrrrr0r0|s 8 rctjsdS|jjdks|jt vrdStstj dddStsMtj d ddd  tjd  ddSd S) 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"rD_get_cudnn_enableddevicetypedtypeCUDNN_TENSOR_DTYPESr0warningswarnr@formatr-sysplatform)tensors r is_acceptablerYs 8 & & ( (u }V##v|;N'N'Nu >>  N    u 77  B I I#6HHLLL"3 J        u 4rnonec tjtjt sdntjtjtjtjddtj f}|tj ||tj ||-t rtj ||tj ||tj||!tjdd||tj||S)Nrall)r"rDrN_get_cudnn_benchmarkr0_cuda_get_cudnn_benchmark_limit_get_cudnn_deterministic_get_cudnn_allow_tf32r_get_cudnn_depthwise_kernel_set_cudnn_enabled_set_cudnn_benchmark_cuda_set_cudnn_benchmark_limit_set_cudnn_deterministic_set_cudnn_allow_tf32r_set_cudnn_depthwise_kernel)_enabled _benchmark_benchmark_limit_deterministic _allow_tf32_fp32_precision_depthwise_kernel orig_flagss r set_flagsrpsn ##%% %%'' NNR(P(P(R(R ))++ &&(( ++FE:: ,,..J ##H--- %%j111# # 001ABBB! )).999 &&{333" ++FE?KKK$ ,,->??? rF Tautoc #RKt5t|||||||}dddn #1swxYwY dVt5t|ddddS#1swxYwYdS#t5t|dddw#1swxYwYwxYwr)rrp)enabled benchmarkbenchmark_limit deterministic allow_tf32fp32_precisiondepthwise_kernelros rflagsr{s ' ( (                            # + , , # # z " " # # # # # # # # # # # # # # # # # # * , , # # z " " # # # # # # # # # # # # # # # #sM377A5 A((A,/A,5B& B B&B B&!B "B&ceZdZeejjejjZeejj ejj Z eejj ejj ZdZer eejjejjZeejjejjZeddZeeddeddZeejjejjZdS) CudnnModuleNrconvr\)__name__ __module__ __qualname__r r"rDrNrbrtr_rerwr]rcrurvr0r^rdr`rfrxrr~rrryrargrzrrrr}r}s-k%(5ux7RSSGK )58+LM  %ux'DIO|~~ %+ H 4 H 4   &(FJ >&& ) )D [""6511""6511N#{ , ,rr}rtrwrurxryrvrz)NNNNNrZN)FFrqFTrZrr)&r+rVrS contextlibrtypingrr"torch.backendsrrrrr r rr torch._Cr ImportErrorrint__annotations__r@r1halffloatdoublerRr0rYrpr{r}modulesrboolstrrrrrs~ %%%%%%  FFF#t""" BBBBL J K L 4    F ####@*<$ CK$98DD H    s<AA