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()
但实际上有可能是无法解决的
,至少对于我来说是这样的 😅文章来源:https://www.toymoban.com/news/detail-651994.html
2. 解决问题 ✅
🚩 导致此问题的原因可能是 指定的GPU的显存不足
,需要将 batch size 改小
文章来源地址https://www.toymoban.com/news/detail-651994.html
到了这里,关于【Error】CUDA error: unknown error的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!