AssertionError: CUDA unavailable, invalid device 0 requested

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

猜你喜欢

转载自blog.csdn.net/li1873997/article/details/133869509
今日推荐