最新消息: USBMI致力于为网友们分享Windows、安卓、IOS等主流手机系统相关的资讯以及评测、同时提供相关教程、应用、软件下载等服务。

For debugging consider passing CUDA_LAUNCH_BLOCKING=1.

业界 admin 26浏览 0评论

环境:

wsl ubuntu22.04

vits2

问题描述:

RuntimeError: CUDA error: unknown error
[rank0]: CUDA kernel errors might be asynchronously reported at some other API call, so the stacktrace below might be incorrect.
[rank0]: For debugging consider passing CUDA_LAUNCH_BLOCKING=1.
[rank0]: Compile with TORCH_USE_CUDA_DSA to enable device-side assertions

解决方案:

1.在train_ms.py开头代码添加

import os
os.environ['CUDA_LAUNCH_BLOCKING'] = '1'

2.重新训练 这边报错消失

环境:

wsl ubuntu22.04

vits2

问题描述:

RuntimeError: CUDA error: unknown error
[rank0]: CUDA kernel errors might be asynchronously reported at some other API call, so the stacktrace below might be incorrect.
[rank0]: For debugging consider passing CUDA_LAUNCH_BLOCKING=1.
[rank0]: Compile with TORCH_USE_CUDA_DSA to enable device-side assertions

解决方案:

1.在train_ms.py开头代码添加

import os
os.environ['CUDA_LAUNCH_BLOCKING'] = '1'

2.重新训练 这边报错消失

发布评论

评论列表 (0)

  1. 暂无评论