search

yasenh / libtorch-yolov5

A LibTorch inference implementation of the yolov5
MIT License
374 stars 114 forks source link

how use this code for CPU-Only? #30

Closed Yasin40 closed 3 years ago

Yasin40 commented 3 years ago

Hi, I wanna use this code to inference with cpu-only, without cuda support?

yasenh commented 3 years ago

@Yasin40 The answer is Yes, you can take a look at the README

Yasin40 commented 3 years ago

@Yasin40 The answer is Yes, you can take a look at the README

I got error when make it without cuda (on system without cuda installed)

Wangbenzhi commented 3 years ago

i think remove "#include <c10/cuda/CUDAStream.h> and include <ATen/cuda/CUDAEvent.h>" in detector.h would work

yasenh commented 3 years ago

@Wangbenzhi thanks for pointing it out, agreed