a dJ @s ddlZddlZddlZddlmZddlZddlmZdZddZ zej dkrej d krej d krej d Z e d ZeD]2Zejer|zeeWq|eyYq|0q|ed ZejedZddZ WneefyYn0ddZddZddZedS)N)warn)_get_extension_pathFcCsdS)NFrrr^/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/torchvision/extension.py_has_opssrnt)r PATH;Z_CTcCsdS)NTrrrrrr'scCststddS)NaCouldn't load custom C++ ops. This can happen if your PyTorch and torchvision versions are incompatible, or if you had errors while compiling torchvision from source. For further information on the compatible versions, check https://github.com/pytorch/vision#installation for the compatibility matrix. Please check your PyTorch version with torch.__version__ and your torchvision version with torchvision.__version__ and verify if they are compatible, and if not please reinstall torchvision so that it matches your PyTorch install.)r RuntimeErrorrrrr_assert_has_ops.src CstsdSddlm}tjj}|dkr|durt|}t|dkr^t|d}t|d}nt|dd}t|d}| d}t|d}t|d }||ks||krt d |d|d |d|d |S) z` Make sure that CUDA versions match between the pytorch install and torchvision install r)cudaNi'r .rzkDetected that PyTorch and torchvision were compiled with different CUDA versions. PyTorch has CUDA Version=z" and torchvision has CUDA Version=zE. Please reinstall the torchvision that matches your PyTorch install.) _HAS_OPSZ torch.versionrtorchopsZ torchvisionZ _cuda_versionstrintsplitr)Ztorch_version_cuda_versionZ tv_versionZtv_majorZtv_minorZ t_versionZt_majorZt_minorrrr_check_cuda_version;s6        rcCs\t|}tjdkrLtjdkrLtjddd}t|drD||ddnt dt j |dS) Nrr z kernel32.dllT)use_last_errorLoadLibraryExWiz)LoadLibraryExW is missing in kernel32.dll) rosnamesys version_infoctypesZWinDLLhasattrrrrr load_library)Zlib_namelib_pathZ _kernel32rrr _load_libraryYs r')r#rr!warningsrrZ_internally_replaced_utilsrrrr r"environZenv_pathrZpath_arrpathexistsZadd_dll_directory Exceptionr&rr% ImportErrorOSErrorrrr'rrrrs6