问题现象描述

• 报错截图

  放大

• 报错文本

  ……
  AssertionError: Torch not compiled with CUDA enabled

原因分析

NPU未使用NPU的流同步方法。

解决措施

使用NPU的流同步方法：

stream = torch_npu.npu.current_stream()
stream.synchronize()