o p i%@svddlmZddlZddlZddlZgZddZddZddZ d d Z d d Z d dZ ddZ ddZdddZdS)) annotationsNcCsvtjjdgddd}tjddgdtjtjjddd }tjdgdd }tjjj |||d }tj |}|||fS) zD Define a simple network composed by a single linear layer. input)Nrfloat32)nameshapedtyperg?) initializer)rrattr)rr)xweightbias) paddlestaticdataZcreate_parameter ParamAttrnnr ConstantZ functionallineartensorsum)rr r linear_outoutra/home/app/PaddleOCR-VL/.venv_paddleocr/lib/python3.10/site-packages/paddle/utils/install_check.py_simple_networks   rcCs*tjddgddggtjd}|dddS)zK Prepare feeding data for simple network. The shape is [1, 2, 2]. ?g@g@g@)rr)nparrayrZreshape)Znp_input_singlerrr _prepare_data,sr!c CTzttjdks JWdSty)}ztd|WYd}~dSd}~ww)z* Check whether CUDA is available. rTzYou are using GPU version PaddlePaddle, but there is no GPU detected on your machine. Maybe CUDA devices is not set properly. Original Error is NF)lenrr cuda_places Exceptionloggingwarningerrr_is_cuda_available6r*c Cr")z) Check whether XPU is available. rTzYou are using XPU version PaddlePaddle, but there is no XPU detected on your machine. Maybe XPU devices is not set properly. Original Error is NF)r#rr xpu_placesr%r&r'r(rrr_is_xpu_availableFr+r-c Cst|r tdn|rtdn |rt|ntdtjdtjjjddd}tjdtjjjd dd}tjjd d ||d }t}t |}||} tj | } | tj jd |d} | dS)z Testing the simple network in dygraph mode using one CPU/GPU/XPU. Args: use_cuda (bool): Whether running with CUDA. use_xpu (bool): Whether running with XPU. ZgpuZxpucpur g?)value)rr rrr) weight_attr bias_attrMbP?Z learning_rate parametersN)rdisable_staticZ set_devicerrr rLinearr!Z to_tensorrrbackward optimizerAdamr5step) use_cudause_xpu use_customcustom_device_namer1r2rZinput_npZ input_tensorrroptrrr_run_dygraph_singleVs4       rAc Cs ttjtjutj}tj}d|_tj||t\}}}tjj ||gdd} Wdn1s>wY|rKt d} n|rSt d} n |r\t |d} nt } tj| } | || j||jti|| dgdWdn1swYtdS)z Testing the simple network with executor running directly, using one CPU/GPU/XPU. Args: use_cuda (bool): Whether running with CUDA. use_xpu (bool): Whether running with XPU. r)Zparameter_listrN)feedZ fetch_list)rZ enable_staticrZ scope_guardZScopeZProgramZ random_seedZ program_guardrZappend_backwardZ CUDAPlaceZXPUPlaceZ CustomPlaceZCPUPlaceExecutorrunrr!r6) r<r=r>r?Z train_progZ startup_progrrr Z param_gradsZplaceexerrr_run_static_single{s:          rFc CsGdddtjj}tj|}t|}tj}tjjd| d}t ddgd}||}t ddgd}|||}| | | dS) z2 train script for parallel training check cs(eZdZdZfddZddZZS)z)train_for_run_parallel..LinearNetz? simple fc network for parallel training check cs.ttjdd|_tjdd|_dS)N r)super__init__rrr7_linear1_linear2)self __class__rrrIs z2train_for_run_parallel..LinearNet.__init__cSs|||S)z! forward )rKrJ)rLr rrrforwardsz1train_for_run_parallel..LinearNet.forward)__name__ __module__ __qualname____doc__rIrO __classcell__rrrMr LinearNets rUr3r4rGrrN)rrZLayer distributedZinit_parallel_envZ DataParallelZMSELossr9r:r5Zrandnr8r;Z clear_grad) rUlayerZdp_layerZloss_fnZadamZinputsZoutputslabelsZlossrrrtrain_for_run_parallels     rYcCstjjtt|ddS)z Testing the simple network in data parallel mode, using multiple CPU/GPU. Args: use_cuda (bool): Whether running with CUDA. use_xpu (bool): Whether running with XPU. device_list (int): The specified devices. )ZnprocsN)rrVZspawnrYr#) device_listrrr _run_parallels r[returnNonec Cstdd}d}d}d}trt}n-trt}n%ttjj dkrAd}ttjj dkrAt dtjj dd|rKd }tj }n+|rUd }tj }n!|rmtjj d}|}tttjj|}n d }tj jdd }t|}t||||t||||td |dz't|dkr|rddl}d|jd<t|td|d|dtdWdSty}zt d|d|dt d|td|d|d}~ww)a Check whether PaddlePaddle is installed correctly and running successfully on your system. Examples: .. code-block:: python >>> import paddle >>> # doctest: +SKIP('the output will change in different run') >>> paddle.utils.run_check() Running verify PaddlePaddle program ... I0818 15:35:08.335391 30540 program_interpreter.cc:173] New Executor is Running. I0818 15:35:08.398319 30540 interpreter_util.cc:529] Standalone Executor is Used. PaddlePaddle works well on 1 CPU. PaddlePaddle is installed successfully! Let's start deep learning with PaddlePaddle now. z(Running verify PaddlePaddle program ... FNrTrzWMore than one kind of custom devices detected, but run check would only be executed on .ZGPUZXPUZCPU) device_countzPaddlePaddle works well on 1 ZxcclZPADDLE_DISTRI_BACKENDzPaddlePaddle works well on  zs.zXPaddlePaddle is installed successfully! Let's start deep learning with PaddlePaddle now.z%PaddlePaddle meets some problem with aqs. This may be caused by: 1. There is not enough GPUs visible on your system 2. Some GPUs are occupied by other process now 3. NVIDIA-NCCL2 is not installed correctly on your system. Please follow instruction on https://github.com/NVIDIA/nccl-tests to test your NCCL, or reinstall it following https://docs.nvidia.com/deeplearning/sdk/nccl-install-guide/index.htmlz Original Error is: z7PaddlePaddle is installed successfully ONLY for single z2! Let's start deep learning with PaddlePaddle now.)printrZis_compiled_with_cudar*Zis_compiled_with_xpur-r#Z frameworkcoreZget_all_custom_device_typer&r'rr$r,listrangeZget_custom_device_countZ cpu_placesrFrAosenvironr[r%) r<r=r>r?Z device_strrZr_rer)rrr run_checkst      rg)r\r]) __future__rr&numpyrr__all__rr!r*r-rArFrYr[rgrrrrs  %&*