YOLOv5-推理时使用GPU

当权重文件训练配置好后，我们打开推理文件detect.py,找到下面这个部分：

图中标红框的语句即是设置检测时是否使用GPU。

1、使用GPU检测，即default设置为0或1,2（取决于电脑GPU块数，不排除有的服务器装了多块GPU）

parser.add_argument('--device', default='0', help='cuda device, i.e. 0 or 0,1,2,3 or cpu')

ps:如果设置GPU报错，可能是pytorch版本装错了，装成CPU版的或与cuda不兼容，则需要重新装pytorch。

2、使用CPU检测，即default设置为CPU

parser.add_argument('--device', default='CPU', help='cuda device, i.e. 0 or 0,1,2,3 or cpu')