yolov5训练时gpu_mem为0

如果在使用 YOLOv5 进行训练时，GPU memory 显示为 0，这可能是由于以下原因：

1. 您的显卡不支持 CUDA。YOLOv5 只能在支持 CUDA 的 NVIDIA 显卡上进行训练。

2. 您的数据集太小或者 batch size 设置过大。如果您的数据集非常小，或者您的 batch size 设置过大，那么您的显存可能会不足以支持模型的训练。尝试减小 batch size 或者增加数据集的大小。

3. 您的显卡内存不足。如果您的显卡内存非常小，可能无法支持 YOLOv5 模型进行训练。尝试使用更大的显卡来进行训练或者减小 batch size。

4. 您没有正确地安装 PyTorch。请确保已经正确地安装了 PyTorch，并且您的 PyTorch 版本与您的 CUDA 版本兼容。

请检查以上原因，看看是否可以解决您的问题。如果您仍然遇到困难，请提供更多信息，以便更好地帮助您。