ubuntu yolov5报错：RuntimeError: CUDA error: no kernel image is available for execution on the device

Plotting labels to runs/train/exp8/labels.jpg…
Traceback (most recent call last):
File “train.py”, line 670, in
main(opt)
File “train.py”, line 565, in main
train(opt.hyp, opt, device, callbacks)
File “train.py”, line 265, in train
check_anchors(dataset, model=model, thr=hyp[‘anchor_t’], imgsz=imgsz)
File “yolov5/utils/autoanchor.py”, line 44, in check_anchors
anchors = m.anchors.clone() * stride # current anchors
RuntimeError: CUDA error: no kernel image is available for execution on the device
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.

原因：安装的cuda版本与torch不兼容
解决方法：pip3 install torch1.9.0+cu111 torchvision0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html