运行测试代码时出错，不能成功执行python detect.py

运行测试代码时(python detect.py --source ./data/images/ --weights weights/yolov5s.pt --conf-thres 0.4)
出现错误：
RuntimeError: CUDA error: CUBLAS_STATUS_INTERNAL_ERROR when calling cublasCreate(handle)
不知道怎么解决，我直接用的百度网盘的代码。

pytorch的版本可能有问题，试试安装17.1版本试试

pip uninstall torch
pip install torch==1.7.1