You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Hi, I think it has nothing to do with device id. You can specify CUDA_LAUNCH_BLOCKING=1 to view the underlying reason. If this cannot give you hint about how to solve it, you may need to follow the tutorials in README.md and Rotate-Tutorial1.ipynb again step by step to see where is the problem.
Hello! When I training your sample dataset, it suffered an error. The printed information is as follow:
Traceback (most recent call last):
File "rotate_train.py", line 557, in
train(hyp, opt, device, tb_writer, rotate=opt.rotate)
File "rotate_train.py", line 318, in train
loss, loss_items = compute_loss(pred, targets.to(device)) # loss scaled by batch_size
File "E:\RotateObjectDetection\rotate-yolov5\utils\loss.py", line 269, in call
iou = rotate_bbox_iou(pbox, tbox[i], CIoU=True, device=device) # iou(prediction, target)
File "E:\RotateObjectDetection\rotate-yolov5\utils\general.py", line 1070, in rotate_bbox_iou
boxes1_xyxyxyxy = xywhrm2xyxyxyxy(boxes1)
File "E:\RotateObjectDetection\rotate-yolov5\utils\general.py", line 985, in xywhrm2xyxyxyxy
xyxyxyxy = torch.matmul(xyxyxyxy, R).view(-1, 8).contiguous()+xywhrm[:, [0, 1, 0, 1, 0, 1, 0, 1]]
RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling
cublasCreate(handle)
I modified the device to cuda "0", it did not work.
The text was updated successfully, but these errors were encountered: