Skip to content
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

GPU 0 is also used when running on other GPU #6007

Open
MoeLau opened this issue Oct 24, 2017 · 0 comments
Open

GPU 0 is also used when running on other GPU #6007

MoeLau opened this issue Oct 24, 2017 · 0 comments

Comments

@MoeLau
Copy link

MoeLau commented Oct 24, 2017

When running the test on 2nd GPU (with id 1) with caffe-RC3, the command nvidia-smi shows that both GPU 0 and 1 are being used. I think the thread for prefetching data (in InternalThread.cpp) uses GPU 0 in default. So how to set GPU ID of the prefetch thread ?
Processes: GPU Memory |
| GPU PID Type Process name Usage |
|=============================================================================|
| 0 28802 C /usr/bin/python 346MiB |
| 2 28802 C /usr/bin/python 7991MiB

Operating system:Ubuntu 14.04
Compiler: GCC
CUDA version (if applicable):8.0
CUDNN version (if applicable):8.0
BLAS:
Python or MATLAB version (for pycaffe and matcaffe respectively):Python

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant