Join GitHub today
GitHub is home to over 20 million developers working together to host and review code, manage projects, and build software together.
New thread which get the forward instance constructed by main thread cannot use GPU #5744
Comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
fengshikun commentedJul 6, 2017
•
edited
Issue summary
I construct a new classification forward instance in main thread like this:
and pass this instance to a new thread to do forward. but this new thread failed to use GPU.
from #4178,
I add
Caffe::set_mode(Caffe::GPU);before each forward operation, but it will complains:How to reset the GPU configuration to make it normal, thanks!
Your system configuration
Operating system: Ubuntu16.06 with CUDA8.0