Namespace(weights=['yolov7.pt'], data='data/coco.yaml', batch_size=32, img_size=640, conf_thres=0.001, iou_thres=0.65, task='val', device='0', single_cls=False, augment=False, verbose=False, save_txt=False,
save_hybrid=False, save_conf=False, save_json=True, project='runs/test', name='yolov7_640_val', exist_ok=False, no_trace=False, v5_metric=False)
Traceback (most recent call last):
File "D:\pythoncode\humanpose\yolo3DPosedemo\yolov7\test.py", line 319, in <module>
test(opt.data,
File "D:\pythoncode\humanpose\yolo3DPosedemo\yolov7\test.py", line 51, in test
device = select_device(opt.device, batch_size=batch_size)
File "D:\pythoncode\humanpose\yolo3DPosedemo\yolov7\utils\torch_utils.py", line 71, in select_device
assert torch.cuda.is_available(), f'CUDA unavailable, invalid device {device} requested' # check availability
AssertionError: CUDA unavailable, invalid device 0 requested
报错原因:没有装gpu得pytorch
解决办法:安装gpu版本得pytorch,问题解决
pip install torch==2.0.0+cu118 torchvision==0.15.1+cu118 torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/cu118