When i run detect.py file of yolov5 just for inference that time show me this error someone help me?

torch.package / torch::deploy
1

RuntimeError: CUDA error: CUBLAS_STATUS_INVALID_VALUE when calling cublasSgemm( handle, opa, opb, m, n, k, &alpha, a, lda, b, ldb, &beta, c, ldc)

2

Could you post a minimal, executable code snippet to reproduce the issue as well as the output of python -m torch.utils.collect_env?

3

ptrblack,

thank you for your time but solved this error and execute the code and it run perfectly