下载
中文
注册
CUDA流同步操作报错

CUDA流同步操作报错

2023/06/06

523

暂无评分
我要评分

问题信息

问题来源产品大类关键字
官方模型训练--

问题现象描述

  • 报错截图

  • 报错文本
    ……
    AssertionError: Torch not compiled with CUDA enabled

原因分析

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

解决措施

使用NPU的流同步方法:

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

本页内容