跑yolov5出现 nan

Question

用cpu跑yolov5时一切正常
用gpu跑yolov5时出现 nan

Answer 1

这可能是PyTorch在其他一些问题中提到的Windows/conda/CUDA11错误，在这种情况下，降级到CUDA 10可以解决这个问题。

或者，您的数据集标签可能存在一些问题。