how to know run to gpu or cpu

walktree / libtorch-yolov3

A Libtorch implementation of the YOLO v3 object detection algorithm

452 stars 131 forks source link

Closed lyxbyr closed 4 years ago

lyxbyr commented 4 years ago

i hava libtorch1.0.0 cuda10 GPU1050Ti, why i test a picture need cost 375ms,I think it run on the cpu thanks!

lyxbyr commented 4 years ago

GPU1050Ti 70+ms