YOLO does not use GPU
zillur-av opened this issue · comments
I think the Yolo implementation here is CPU-based while PyTorch is cuda-based. How can I use Cuda even for Yolo detections?
PyTorch implementation for 3D Bounding Box Estimation Using Deep Learning and Geometry
zillur-av opened this issue · comments
I think the Yolo implementation here is CPU-based while PyTorch is cuda-based. How can I use Cuda even for Yolo detections?