o ְiS @s0dZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl m Z m Z mZmZmZddlZdadadadada daddddZejdddkZd ed dfd d Zd dZddZddZ ddZ!ddZ"ddZ#ddZ$ddZ%ddZ&dd Z'd!d"Z(d#d$Z)d%d&Z*d ee ee ffd'd(Z+d)ed ee ee ffd*d+Z,d,ed e-fd-d.Z.d/d0Z/d1ed ee0e0e0ffd2d3Z1d4ed1ed5ed eej2j3fd6d7Z4d1ed5ed e5efd8d9Z6d1ed5ed e5efd:d;Z7de0d?e0d eefd@dAZ9dBdCZ:d efdDdEZ;d efdFdGZzinclude/hipcubz) cupy_backends.cuda.apirBr/r2r6abspath__file__is_hipisdirr=r$)rB current_dirr(r@rrrr)s$     r)cCsxtjdrt}|duo|ddk}t}|dur'|r|}ntj|d}n d}|s0t dt d |tjt ddd }tj |rOt d |n t d |d}d tjkr|durt d |t|tj|d}tj|rt d|t||durt d |t|dSdS|durt d |tjdd}|tj|tjd<dSdSdS)Nwin32 packagingcondar;zZCUDA path could not be detected. Set CUDA_PATH environment variable if CuPy fails to load.z CUDA_PATH: {}cupy.datalibzWheel shared libraries: {}z%Not wheel distribution ({} not found))zAdding DLL search path: {}x64z&Adding DLL search path (for CUDA 13): zAdding to PATH: {}PATHr)rplatform startswithget_preload_configrr/r2r=warningswarnrformatget_cupy_install_pathrG version_infoadd_dll_directoryr3r0r1pathsep)configis_condar7 cuda_bin_path wheel_libdircuda_bin_x64_pathr2rrr_setup_win32_dll_directorysV       -rbcCstjtjtjtdS)Nz..)r/r2rDr=r6rErrrrrYsrYcCs.tjdd}|durtjdStj|S)a|Returns the directory where CUDA external libraries are installed. This environment variable only affects wheel installations. Shared libraries are looked up from `$CUPY_CUDA_LIB_PATH/$CUDA_VER/$LIB_NAME/$LIB_VER/{lib,lib64,bin}`, e.g., `~/.cupy/cuda_lib/11.2/cudnn/8.1.1/lib64/libcudnn.so.8.1.1`. The default $CUPY_CUDA_LIB_PATH is `~/.cupy/cuda_lib`. CUPY_CUDA_LIB_PATHNz~/.cupy/cuda_lib)r/r0r1r2 expanduserrD)cupy_cuda_lib_pathrrrget_cupy_cuda_lib_paths   rfcCstdurtdatS)Nz _wheel.json)_preload_config_get_json_datarrrrrUsrUnamecCsZtjtdd|}tj|sdSt| }t|WdS1s&wYdS)NrLrM)r/r2r=rYr3openjsonload)ri config_pathfrrrrhs   $rhrNcCst}|dus |ddkrtd|ddS|tvr"td|||vr0td|d dSt|dur?td |dSd S) z(Returns if the preload can be attempted.NrJrKzNot preloading z( as this is not a pip wheel installationFzUnknown preload library: zPreload z not configured in wheelzPreload already attempted: T)rUr _preload_libsAssertionError)rNr]rrr_can_attempt_preload"s rqc s tdts dSit<td}td|ttddd}|D]։tdd d d fd d dD}dkrhd}t|d|}ndkr{d}t|d|}|D]S}tj |std|q}ztd|t |t|<tdWnht y}z t|j}d|d|d |d }t|t|WYd}~q}d}~wwtdtddzt t<tdWq7t y }ztdd|WYd}~q7d}~wwq7dS)zPreload dependent shared libraries. The preload configuration file (cupy/.data/_wheel.json) will be added during the wheel build process. z"Preloading triggered for library: Ncudaz$CuPy wheel package built for CUDA {}zCuPy CUDA library directory: {}version filenames Looking for z version  ()c s&g|]}tjd|qS)rr)r/r2r=).0xr]refilenamerNrsrr Zsz$_preload_library..)rNlib64r;r min_pypi_versionr z"Rejected candidate (not found): {}zTrying to load Loadedz CuPy failed to preload library (z): zFile {} could not be foundz from default search pathzLibrary z could not be preloaded: )rrqrorUrXrf_get_cutensor_from_wheel_get_nccl_from_wheelr/r2r3ctypesCDLL Exceptiontype__name__rVrW) rN cuda_versionrt libpath_candsr~libpathee_typerrrzr_preload_library>sx          rrscCs^tjd|dd}t|d}t|dkrt|dnd}t|dkr(t|dnd}|||fS)Nz[^\d]rO)maxsplitr)resplitintlen)rspartsmajorminorpatchrrr_parse_versions   rpackage_prefixrrc Cs|d\}}|d|}ztj|}Wntjjy,t|d|dYdSwt|j}t|}|d|dkoU|d|dkpU|d|dkoU|d|dk} | sjt|d|d |d |jdSt|d|d |j|S) z Returns the distribution of the given package name and version installed via pip (e.g., cutensor-cuXX). If the package is not found or incompatible, returns None. .z-cuz wheel package (z) not installedNrrrz) incompatible: expected z, found z ) found: )r importlibmetadata distributionPackageNotFoundErrorrrrs) rrsrrcuda_major_ver_pkgdistactualexpected is_compatiblerrr_find_compatible_wheels. rcCsRtd||}|dur gStjdkr|d|dd}n|d}t|gS)zx Returns the list of shared library path candidates for cuTENSOR installed via Pip (cutensor-cuXX package). r Nlinuxzcutensor/lib/libcutensor.so.rrzcutensor\bin\cutensor.dll)rrrS locate_filerstr)rsrrr shared_librrrrs    rcCsPtjdkrgStd||}|durgS|d|ddg}dd|DS) zw Returns the list of shared library path candidates for NCCL installed via Pip (nvidia-nccl-cuXX package). rz nvidia-ncclNznvidia/nccl/lib/libnccl.so.rrcSsg|]}t|qSr)r)rxrNrrrr|sz(_get_nccl_from_wheel..)rrSrrr)rsrrr shared_libsrrrrs  rc Cst}|dus ||vr dS|ddkre|d}|dd}|dkr=|dd}t|d}d|d |d |d d }n6|d kr\|d d}t|d}d|d |d |d d }nd|d|}n|ddkrqd|}nttd|dt|jdt|d|d dS)NrJpiprrrrr r~zpip install "cutensor-cuz>=z, cupy-cuda80 cupy-cuda90 cupy-cuda91 cupy-cuda92 cupy-cuda100 cupy-cuda101 cupy-cuda102 cupy-cuda110 cupy-cuda111 cupy-cuda112 cupy-cuda113 cupy-cuda114 cupy-cuda115 cupy-cuda116 cupy-cuda117 cupy-cuda118 cupy-cuda11x cupy-cuda12x cupy-cuda13x cupy-rocm-4-0 cupy-rocm-4-1 cupy-rocm-4-2 cupy-rocm-4-3 cupy-rocm-5-0rLcSsh|] }|jddqS)NameN)rr1)rxdrrr Osz1_detect_duplicate_installation..rz, z -------------------------------------------------------------------------------- CuPy may not function correctly because multiple CuPy packages are installed in your environment: a/ Follow these steps to resolve this issue: 1. For all packages listed above, run the following command to remove all existing CuPy installations: $ pip uninstall If you previously installed CuPy via conda, also run the following: $ conda uninstall cupy 2. Install the appropriate CuPy package. Refer to the Installation Guide for detailed instructions. https://docs.cupy.dev/en/stable/install.html -------------------------------------------------------------------------------- )rr distributionsrr=sortedrVrW)knowninstalled_namescupy_installedcupy_packages_listrrr_detect_duplicate_installation,s! rc Csdd}tjdkr0z|t7}W|Sty/}z|dt|jd|7}WYd}~|Sd}~ww|S)NaFailed to import CuPy. If you installed CuPy via wheels (cupy-cudaXXX or cupy-rocm-X-X), make sure that the package matches with the version of CUDA or ROCm installed. On Linux, you may need to set LD_LIBRARY_PATH environment variable depending on how you installed CUDA/ROCm. On Windows, try setting CUDA_PATH environment variable. Check the Installation Guide for details: https://docs.cupy.dev/en/latest/install.htmlrIz% The cause could not be identified: : )rrS_diagnose_win32_dll_loadrrr)rrrrr_diagnose_import_errorps"  rc Cs"td}|dur dSddlm}td}|j|j|jg|j_|j|j_ dddt dg}t d}|d D]T}zt|}||j |t ||d |d |jWq7tyh|d |d Yq7ty}z|d |d t|jd|dWYd}~q7d}~wwd|S)Nz _depends.jsonrr)wintypeskernel32z CUDA Path: zDLL dependencies:idependsz z -> z -> not foundz -> error (rrwr)rhrrrHANDLELPWSTRDWORDGetModuleFileNameWargtypesrestypercreate_unicode_buffer_handlerappendvalueFileNotFoundErrorrrrr=)rrrlinesfilepathridllrrrrrs4         0 r)=__doc__rimportlib.metadatarrkr/os.pathrSrr4rtypingrrrrrrVrrr"r%r(rgror0r1rrrrr!r$r'r*rr r#r&r)rbrYrfrUrhboolrqrrrr Distributionrlistrrrrrrrrrrrsv 6 M  ! 5D