-
Notifications
You must be signed in to change notification settings - Fork 176
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
yolo2 and deepsort models run on the same gpu? #26
Comments
you can change the GPU Flag in YOLO :
by default it's on GPU 0 , but you can set it to another GPU (for example /gpu:1) , hopefully deep_sort will still run on GPU 0 (I only have one GPU so can't test it ) Another thing to try is to set deep sort to run on GPU 1 , by adding this line to deep sort tf_session :
these should replace line 318 in the script if it works out, a pull request would be appreciated for future users :) |
thanks for reply! |
I tried the two ways you mentioned, however it both turns out that they have no effects on the gpu assignment. |
try using |
it seems that in this project yolo2 and deepsort models run on the same gpu, can it be configured to run different models on multiple different gpus to further improve the fps of the whole end-to-end system?
The text was updated successfully, but these errors were encountered: