Traceback (most recent call last):
File "flask_app.py", line 21, in
device = select_device(opt['device'])
File "/home/yigou/yolov5/utils/torch_utils.py", line 65, in select_device
assert torch.cuda.is_available(), 'CUDA is not available, use --device cpu or do not pass a --device'
AssertionError: CUDA is not available, use --device cpu or do not pass a --device
assert torch.cuda.is_available()
在你的代码中强制要求使用cuda，可以注释掉后使用