环境信息收集工具

环境信息收集工具为用户提供当前训练设备的软硬件信息,可用于问题排查、环境确认。用户可通过运行pytorch/torch_npu/utils/collect_env.py脚本,收集当前环境信息,例如:CANN版本与驱动版本信息、PyTorch框架版本、CPU信息、OS信息等。

  1. 准备运行环境,可参见CANN软件安装指南完成依赖软件安装与环境变量配置并安装PyTorch框架及插件。
  2. 执行命令,开始收集环境信息。

    python  torch_npu/utils/collect_env.py

    命令执行完成后,会将环境信息全部打屏,信息内容可参见表1

    表1 可收集信息

    信息内容

    说明

    PyTorch version: {torch_version}

    PyTorch框架版本号。

    Torch-npu version: {torch_npu_version}

    Torch_npu插件版本号。

    Is debug build: {is_debug_build}

    是否为debug编译。

    OS: {os}

    操作系统。

    GCC version: {gcc_version}

    GCC版本号。

    Clang version: {clang_version}

    Clang版本号。

    CMake version: {cmake_version}

    CMkae版本号。

    Libc version: {libc_version}

    Libc版本号。

    Python version: {python_version}

    Python版本号。

    Python platform: {python_platform}

    系统系统架构。

    Is XNNPACK available: {is_xnnpack_available}

    XNNPACK是否可用。

    CPU:{cpu_info}

    CPU信息。

    CANN:{cann_version}

    CANN版本号。

    CANN driver:{cann_driver_version}

    CANN驱动版本号。

    NPU mapping info:{npu_mapping_info}

    npu-smi info。

    NPU count info:{npu_count_info}

    NPU个数。

    Versions of relevant libraries:{pip_packages}、{conda_packages}

    pip与conda环境软件包信息。