环境信息收集工具
环境信息收集工具为用户提供当前训练设备的软硬件信息,可用于问题排查、环境确认。用户可通过运行pytorch/torch_npu/utils/collect_env.py脚本,收集当前环境信息,例如:CANN版本与驱动版本信息、PyTorch框架版本、CPU信息、OS信息等。
- 准备运行环境,可参见《CANN 软件安装指南》完成依赖软件安装与环境变量配置并安装PyTorch框架及插件。
- 执行命令,开始收集环境信息。
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环境软件包信息。