训练yolov5,cpu能正常训练，而gpu训练就会输出nan

上面分别是CPU 运行yoloV5和GPU运行时出现的结果，cpu运行时正常，gpu就出现nan

看下版本是否匹配，cuda版本和pytorch版本。
cuda版本：
打开cmd，输入

nvcc --version

Windows查看pytorch的版本

import torch
print(torch.__version__) #注意是双下划线

然后发出来