CUDA流同步操作报错

问题描述

• 报错截图

  

• 报错文本

  ……
  AssertionError: Torch not compiled with CUDA enabled

问题分析

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

处理方法

使用NPU的流同步方法：

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