Traceback (most recent call last): File "f:/YOLO/YOLO/yolov5-6.0/gen_wts.py", line 29, in <module> device = select_device('gpu') File "f:\YOLO\YOLO\yolov5-6.0\utils\torch_utils.py", line 67, in select_device assert torch.cuda.is_available(), f'CUDA unavailable, invalid device {device} requested' # check availability AssertionError: CUDA unavailable, invalid device gpu requested

这个错误提示是因为您在使用 GPU 运行程序时，CUDA 不可用。可能原因如下：

1. 您的电脑没有安装 CUDA 或者 CUDA 版本不符合要求。
2. 您的电脑没有安装相应的显卡驱动程序。
3. 您的显卡不支持 CUDA。

您可以尝试以下解决方案：

1. 确认您的电脑是否安装了 CUDA，如果没有安装，请到 NVIDIA 官网下载安装。
2. 确认您的电脑是否安装了相应的显卡驱动程序，如果没有安装，请到显卡厂商官网下载安装。
3. 确认您的显卡是否支持 CUDA，如果不支持，那么您只能使用 CPU 运行程序。

如果您确认以上三个问题都不存在，那么您可以尝试在程序开头添加以下代码，手动指定可用设备：

import torch

device = torch.device('cuda') if torch.cuda.is_available() else torch.device('cpu')
print('Using device:', device)

这段代码会自动检测 CUDA 是否可用，如果可用则使用 GPU 运行程序，否则使用 CPU 运行程序。