Run models on CPU?
vedantroy opened this issue · 1 comments
vedantroy commented
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
for anyone looking now: set MODEL.DEVICE cpu
(in the --options flag if you are running demo/demo.py)