/j jddlZddlZddlmZdZdZedrdZdZd ZedS) N)_get_extension_pathc H t|}tj|dS#tt f$r^}t jdr4ddl }| d|dt|j d|Yd}~dSd}~wwxYw) zLoad a library, optionally warning on failure based on env variable. Returns True if the library was loaded successfully, False otherwise. T-TORCHVISION_WARN_WHEN_EXTENSION_LOADING_FAILSrNzFailed to load 'z ' extension: z: F) rtorchops load_library ImportErrorOSErrorosenvirongetwarningswarntype__name__)lib_namelib_pathers Z/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/torchvision/extension.py _load_libraryrs &x00 x(((t  ! :>>I J J ] OOO MM[X[[DGGDT[[XY[[ \ \ \uuuuu s.2B!ABB!cdS)NFrr_has_opsrs 5r_CcdS)NTrrrrrrstrc@tstddS)NagCouldn'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. Set TORCHVISION_WARN_WHEN_EXTENSION_LOADING_FAILS=1 and retry to get more details.)r RuntimeErrorrrr_assert_has_opsr #s0 ::   a      rc tsdSddlm}tjj}|dkr|t|}t|dksJd|dt|dd}t|d }| d }t|d}t|d }||krtd |d |d |d |d |S)z` Make sure that CUDA versions match between the pytorch install and torchvision install r)cudaNi.zUnexpected CUDA version z, please file a bug report..rzqDetected that PyTorch and torchvision were compiled with different CUDA major versions. PyTorch has CUDA Version=z" and torchvision has CUDA Version=zE. Please reinstall the torchvision that matches your PyTorch install.) r torch.versionr#rr torchvision _cuda_versionstrintsplitr)torch_version_cuda_version tv_versiontv_majortv_minor t_versiont_majort_minors r_check_cuda_versionr51sI ::r888888y$2244H2~~,8]] :%''')iH)i)i)i'''z!A#''z!}%%&,,S11 il##il## h  V,3VV6=VV (VV+3VVV  Or)r r_internally_replaced_utilsrrrr r5rrrr7s ;;;;;;"=    6r