Join GitHub today
GitHub is home to over 28 million developers working together to host and review code, manage projects, and build software together.Sign up
Predictor interface - GPU #799
As referenced in many other questions (#323 #503 etc.), the Predictor interface does not by default have GPU support. How can you run prediction on a pretrained network (such as bvlc_alexnet) using GPU?
I tried the following code:
INIT_NET, PREDICT_NET, and img are set in the same way as in the tutorial.
But I'm getting the following error:
For some reason there's a problem parsing the .pb file, but the Predictor interface worked totally fine without using this parsing. What am I doing wrong here? Is there a better way to do this? As many others have mentioned, the Predictor interface should definitely have an option to use GPU support.
Running on Windows 10, CUDA 8.0, cuDNN 6.0
Got it working by changing