o Qeà(ã@s|ddlmZddlZddlZddlZddlZgZdd„Zdd„Z dd„Z d d „Z d d „Z d d„Z dd„Zdd„Zdd„ZdS)é)Úabsolute_importNcCsvtjjdgd¢dd}tjddgdtjtjj d¡dd }tjdgdd }tjjj |||d }tj   |¡}|||fS) zD Define a simple network composed by a single linear layer. Úinput)NérÚfloat32)ÚnameÚshapeÚdtyperégš™™™™™¹?)Ú initializer)rrÚattr)rr)ÚxÚweightÚbias) ÚpaddleÚstaticÚdataZcreate_parameterÚ ParamAttrÚnnr ÚConstantZ functionalÚlinearÚtensorÚsum)rr rÚ linear_outÚout©rúJD:\Projects\ConvertPro\env\Lib\site-packages\paddle/utils/install_check.pyÚ_simple_networksþý  rcCsjtjddgddggtjd}|dkr| |dd¡Sg}t|ƒD]}| |¡qt |¡}| |dd¡}|S)z“ Prepare feeding data for simple network. The shape is [device_count, 2, 2]. Args: device_count (int): The number of devices. çð?g@g@g@)rér)ÚnpÚarrayrZreshapeÚrangeÚappend)Ú device_countZnp_input_singleZ input_listÚiZ np_input_mutirrrÚ _prepare_data+s   r%c CóTzttj ¡ƒdks J‚WdSty)}zt d |¡¡WYd}~dSd}~ww)z) Check whether CUDA is avaiable. rTz“You 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Ú ExceptionÚloggingÚwarningÚformat©ÚerrrÚ_is_cuda_available?óý€ûr/c Cr&)z( Check whether NPU is avaiable. rTz’You are using NPU version PaddlePaddle, but there is no NPU detected on your machine. Maybe NPU devices is not set properly. Original Error is {}NF)r'rrÚ npu_placesr)r*r+r,r-rrrÚ_is_npu_availableNr0r2c Cr&)z( Check whether XPU is avaiable. rTz’You 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,r-rrrÚ_is_xpu_available]r0r4c CsØt ¡|r t d¡n|rt d¡n |rt d¡nt d¡tjdtjjjddd}tjd tjjjd dd}tjjd d ||d }tdƒ}t  |¡}||ƒ}tj   |¡} |   ¡tj jd| ¡d} |  ¡dS)zò Testing the simple network in dygraph mode using one CPU/GPU/XPU/NPU. Args: use_cuda (bool): Whether running with CUDA. use_xpu (bool): Whether running with XPU. use_npu (bool): Whether running with NPU. ZgpuZxpuZnpuÚcpur gà?)Úvalue)rr rrré)Ú weight_attrÚ bias_attrrgü©ñÒMbP?)Ú learning_rateÚ parametersN)rÚdisable_staticZ set_devicerrr rZLinearr%Z to_tensorrrZbackwardÚ optimizerZAdamr;Ústep) Úuse_cudaÚuse_xpuÚuse_npur8r9rZinput_npZ input_tensorrrÚoptrrrÚ_run_dygraph_singlels8     ÿÿý  ÿ rCc Cs&t ¡tj tj ¡¡xtj ¡}tj ¡}d|_tj ||¡tƒ\}}}tjj ||j gdd}Wdƒn1s?wY|rLt  d¡} n|rTt  d¡} n |r\t  d¡} nt ¡} tj | ¡} |  |¡| j||j tdƒi|j |dj gdWdƒn1sˆwYt ¡dS)a Testing the simple network with executor running directly, using one CPU/GPU/XPU/NPU. Args: use_cuda (bool): Whether running with CUDA. use_xpu (bool): Whether running with XPU. use_npu (bool): Whether running with NPU. r)Zparameter_listrN©ÚfeedZ fetch_list)rÚ enable_staticrÚ scope_guardÚScopeÚProgramZ random_seedÚ program_guardrZappend_backwardrÚ CUDAPlaceÚXPUPlaceÚNPUPlaceÚCPUPlaceÚExecutorÚrunr%r<) r?r@rAÚ train_progÚ startup_progrrr Z param_gradsÚplaceÚexerrrÚ_run_static_singles8    ÿÿþ      þì rUc CsJt ¡tj tj ¡¡Štj ¡}tj ¡}tj ||¡!tƒ\}}}tj  |¡} d| _ tj j dd  | ¡Wdƒn1sCwYtj |¡j| j|d} |r\t d¡} n|rft d¡} |} n|rpt d¡} |} nt ¡} tj | ¡} |  |¡| j| |jtt|ƒƒi| jgdWdƒn1sšwYt ¡dS)a( 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. use_npu (bool): Whether running with NPU. device_list (int): The specified devices. Tg{®Gáz„?)r:N)Z loss_nameZplacesrrD)rrFrrGrHrIrJrrÚmeanZ persistabler=ZSGDZminimizeZCompiledProgramZwith_data_parallelrrKrLrMrNrOrPr%r'r<) r?r@rAÚ device_listrQrRrrÚ_ZlossZ compiled_progrSrTrrrÚ_run_static_parallel´sB     üÿþ     þæ rYc CsBtdƒd}d}d}t ¡rtƒ}nt ¡rtƒ}nt ¡r!tƒ}|r+d}tj  ¡}n|r5d}tj  ¡}n|r?d}tj  ¡}n d}tjj dd}t |ƒ}t|||ƒt|||ƒtd  |¡ƒzt||||ƒtd  ||¡ƒtd ƒWdSty }zt d  ||¡¡t d  |¡¡td |¡ƒWYd}~dSd}~ww)a× Check whether PaddlePaddle is installed correctly and running successfully on your system. Examples: .. code-block:: python import paddle paddle.utils.run_check() # Running verify PaddlePaddle program ... # W1010 07:21:14.972093 8321 device_context.cc:338] Please NOTE: device: 0, CUDA Capability: 70, Driver API Version: 11.0, Runtime API Version: 10.1 # W1010 07:21:14.979770 8321 device_context.cc:346] device: 0, cuDNN Version: 7.6. # PaddlePaddle works well on 1 GPU. # PaddlePaddle works well on 8 GPUs. # PaddlePaddle is installed successfully! Let's start deep learning with PaddlePaddle now. z(Running verify PaddlePaddle program ... FZGPUZXPUZNPUZCPUr)r#z PaddlePaddle works well on 1 {}.z"PaddlePaddle works well on {} {}s.zXPaddlePaddle is installed successfully! Let's start deep learning with PaddlePaddle now.a›PaddlePaddle meets some problem with {} {}s. 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: {}zkPaddlePaddle is installed successfully ONLY for single {}! Let's start deep learning with PaddlePaddle now.N)ÚprintrZis_compiled_with_cudar/Zis_compiled_with_xpur4Zis_compiled_with_npur2rr(r3r1Z cpu_placesr'rUrCr,rYr)r*r+)r?r@rAZ device_strrWr#r.rrrÚ run_checkßs\     ÿ ÿúÿÿ€ör[)Ú __future__rÚosr*ÚnumpyrrÚ__all__rr%r/r2r4rCrUrYr[rrrrÚs $$ +