在模型运行或者算子运行时遇到报错“Error in atexit._run_exitfuncs:”

问题描述

• 报错截图

  

• 报错文本

  Error in atexit._run_exitfuncs:
  Traceback (most recent call last):
    File "/root/archiconda3/envs/***/lib/python3.7/site-packages/torch/__init__.py", line 429, in _npu_shutdown
      torch._C._npu_shutdown()
  RuntimeError

问题分析

在torch初始化时，若未通过torch_npu.npu.device(id)指定NPU设备，则默认使用device 0设备。若直接使用其他NPU设备，如指定在device 1上创建tensor，那么在运行时会出现上述错误。

处理方法

在调用NPU设备之前，通过torch_npu.npu.set_device(device)指定需要使用的NPU设备即可。