CenterTrack icon indicating copy to clipboard operation
CenterTrack copied to clipboard

Published 20 hours ago • verified publisher icon xingyizhou

in cpu?

Open jinzhenmu opened this issue 4 years ago • 3 comments

  • hi

  • when python3 demo.py tracking,ddd --load_model ../models/nuScenes_3Dtracking.pth --dataset nuscenes --pre_hm --track_thresh 0.1 --demo ../videos/nuscenes_mini.mp4

  • AssertionError: The NVIDIA driver on your system is too old (found version 9000). Please update your GPU driver by downloading and installing a new version from the URL: http://www.nvidia.com/Download/index.aspx Alternatively, go to: https://pytorch.org to install a PyTorch version that has been compiled with your version of the CUDA driver.

  • This code can be used in cpu?

jinzhenmu avatar Apr 04 '20 13:04 jinzhenmu

  • in src/lib/detertor.py

  • #torch.cuda.synchronize()

  • it works

jinzhenmu avatar Apr 04 '20 13:04 jinzhenmu

The current version does not work on CPU due to the use of DCN layers.

xingyizhou avatar Apr 12 '20 20:04 xingyizhou

@xingyizhou I am also interested in doing benchmark on CPU. It would be great to have the CPU version.

bryanbocao avatar Feb 24 '22 01:02 bryanbocao