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

Does ocrd_calamari use GPU using ocrd/all:maximum-cuda image? #68

Closed
mikegerber opened this issue Oct 11, 2021 · 6 comments
Closed

Does ocrd_calamari use GPU using ocrd/all:maximum-cuda image? #68

mikegerber opened this issue Oct 11, 2021 · 6 comments
Assignees
Labels
bug Something isn't working question Further information is requested

Comments

@mikegerber
Copy link
Collaborator

mikegerber commented Oct 11, 2021

Reported by @jbarth-ubhd in Gitter (Oct 06 10:50)

calamari-recognize does not run on GPU, even with maximum-cuda

This seems to be concluded from this memory usage graph:

image
gpu memory usage, red=sbb-binarize, blue=eynollah-segment, green=calamari-recognize

@mikegerber mikegerber added bug Something isn't working question Further information is requested labels Oct 11, 2021
@mikegerber
Copy link
Collaborator Author

I suspect that it's actually running on GPU but does not use much memory compared to eynollah or sbb-binarize, but I'll investigate anyway to confirm.

@maxnth
Copy link

maxnth commented Oct 12, 2021

I can't comment on whether ocrd_calamari actually leverages the GPU or not but AFAIK calamari should seldomly put heavy load on the GPU during recognition anyways.

@mikegerber
Copy link
Collaborator Author

Confirmed: It's not using GPU using the ocrd/all:maximum-cuda image. I'm investigating further. @jbarth-ubhd

@mikegerber
Copy link
Collaborator Author

TensorFlow and CUDA, the endless joy:

(headless-tf2) root@29b1f018638e:/usr/local/sub-venv/headless-tf2# pip list | grep tensorflow
tensorflow               2.6.0

(headless-tf2) root@29b1f018638e:/usr/local/sub-venv/headless-tf2# find /usr/ -name "libcuda.so*"
/usr/local/cuda-10.0/compat/libcuda.so.1
/usr/local/cuda-10.0/compat/libcuda.so
/usr/local/cuda-10.0/compat/libcuda.so.410.129

So this is CUDA Toolkit 10.0 but PyPi's TensorFlow 2.6.0 requires CUDA Toolkit 11.1 or 11.2.1
(see https://github.com/mikegerber/test-nvidia README for the full list of compatible versions.)

I believe @bertsky was already working on putting more than one CUDA Toolkit version into one image?

@mikegerber
Copy link
Collaborator Author

OCR-D/ocrd_all#263

@mikegerber mikegerber changed the title Does ocrd_calamari use GPU using maximum-cuda image? Does ocrd_calamari use GPU using ocrd/all:maximum-cuda image? Oct 26, 2021
@mikegerber
Copy link
Collaborator Author

mikegerber commented Oct 26, 2021

AFAICS It's a problem in ocrd_all, not here. I'm copying my comment above to OCR-D/ocrd_all#279 and closing this.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
bug Something isn't working question Further information is requested
Projects
None yet
Development

No branches or pull requests

2 participants