【Error】CUDA error: unknown error

1. 问题描述

依然是在 wsl2 下的 ubuntu20.04 上运行 YOLOv8 出现了以下错误

CUDA kernel errors might be asynchronously reported at some other API call,so the stacktrace below might be incorrect.
For debugging consider passing CUDA_LAUNCH_BLOCKING=1.

网上搜索的，得到的结果可能是说 torch、torchvision、cuda版本不对应导致，并采用下列方式解决

torch.cuda.current_device()

但实际上有可能是无法解决的，至少对于我来说是这样的

2. 解决问题 ✅

导致此问题的原因可能是 指定的GPU的显存不足，需要将 batch size 改小