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

VIAME appears to be using wrong GPU #175

Open
gcutt opened this issue Mar 1, 2024 · 0 comments
Open

VIAME appears to be using wrong GPU #175

gcutt opened this issue Mar 1, 2024 · 0 comments

Comments

@gcutt
Copy link

gcutt commented Mar 1, 2024

VIAME Desktop, via DIVE Desktop GUI, tried to train resnet and efficientnet classifier models on an annotated image dataset. Resulted in errors from both jobs, maybe related to insufficient memory (copied below).

More concerning - it is clear that VIAME is not using the NVIDIA GPU (GPU 1), the GPU monitor reports no applications running on the GPU, but there is activity on the Intel Graphics GPU (GPU 0).

How do I specify for VIAME to use GPU 1 (the NVIDIA GPU) instead of GPU 0 for model training? And, how to check and verify which GPU is actually being used?

Thank you.

Error from log --
[--
File "X:\VIAME\lib\python3.6\site-packages\torch_init_.py", line 124, in
raise err
OSError: [WinError 1455] The paging file is too small for this operation to complete. Error loading "X:\VIAME\lib\python3.6\site-packages\torch\lib\caffe2_detectron_ops_gpu.dll" or one of its dependencies.
Traceback (most recent call last):
--]

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