环境信息收集工具为用户提供当前训练设备的软硬件信息,可用于问题排查、环境确认。用户可通过运行pytorch/torch_npu/utils/collect_env.py脚本,收集当前环境信息,例如:CANN版本与驱动版本信息、PyTorch框架版本、CPU信息、OS信息等。
python torch_npu/utils/collect_env.py
命令执行完成后,会将环境信息全部打屏,信息内容可参见表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环境软件包信息。 |