search

aim-uofa / AdelaiDet

AdelaiDet is an open source toolbox for multiple instance-level detection and recognition tasks.
https://git.io/AdelaiDet
Other
3.36k stars 645 forks source link

Run models on CPU? #571

Open vedantroy opened 2 years ago

vedantroy commented 2 years ago

Would it be possible to run the models on the CPU, in order to avoid the error:

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

drinking-code commented 5 months ago

for anyone looking now: set MODEL.DEVICE cpu (in the --options flag if you are running demo/demo.py)