RuntimeError: Found no NVIDIA driver

Hi, i tried python demo.py --source data/example.jpg but i ran into following error:

File "/home/yiwei/dev_ws_ml/src/YOLOPv2/demo.py", line 50, in detect
    model  = torch.jit.load(weights)
  File "/home/yiwei/miniconda3/envs/YOLOPv2/lib/python3.8/site-packages/torch/jit/_serialization.py", line 161, in load
    cpp_module = torch._C.import_ir_module(cu, f, map_location, _extra_files)
RuntimeError: Found no NVIDIA driver on your system. Please check that you have an NVIDIA GPU and installed a driver from http://www.nvidia.com/Download/index.aspx

i only want to try this example image with my cpu and i do not have a GPU installed.

can anyone give me some help?

[Miaoishappening]

@sgzqc Hi, Thanks for replying! That really helped.

With this torch.jit.load(weights, map_location='cpu') it works fine.

And i also changed the default --device to “cpu”

the output still warns me, but delivers the result.

Namespace(agnostic_nms=False, classes=None, conf_thres=0.3, device='cpu', exist_ok=False, img_size=640, iou_thres=0.45, name='exp', nosave=False, project='runs/detect', save_conf=False, save_txt=False, source='data/example.jpg', weights='data/weights/yolopv2.pt')
/home/yiwei/miniconda3/envs/YOLOPv2_t/lib/python3.7/site-packages/torch/cuda/__init__.py:52: UserWarning: CUDA initialization: Found no NVIDIA driver on your system. Please check that you have an NVIDIA GPU and installed a driver from http://www.nvidia.com/Download/index.aspx (Triggered internally at  /pytorch/c10/cuda/CUDAFunctions.cpp:100.)
  return torch._C._cuda_getDeviceCount() > 0
/home/yiwei/miniconda3/envs/YOLOPv2_t/lib/python3.7/site-packages/torch/nn/functional.py:3063: UserWarning: Default upsampling behavior when mode=bilinear is changed to align_corners=False since 0.4.0. Please specify align_corners=True if the old behavior is desired. See the documentation of nn.Upsample for details.
  "See the documentation of nn.Upsample for details.".format(mode))
384x640 Done. (1.116s)
 The image with the result is saved in: runs/detect/exp8/example.jpg
inf : (1.1156s/frame)   nms : (0.0020s/frame)
Done. (1.269s)

[sgzqc]

Maybe you can try this: torch.jit.load(buffer, map_location='cpu')