DDMAL / DDMAL/Rodan

Text Alignment not using GPU properly

Open
#1,308 2 comments 0 reactions 0 assignees View on GitHub
Dominant language
CSS
Stars
46
Forks
13
PR merge metrics
No merged PRs in 30d

Description

I was testing Text Alignment on Production (and also for `fix1305`) and noticed that the job was not using NVIDIA GPU, even with proper setup. `nvidia-smi` shows nothing.

Here's the log from `make gpu-celery_log` on Production server:

```
[2025-06-27 13:39:44,017: INFO/MainProcess] Received task: Text Alignment[4366a2d8-1385-40f1-abd2-410ed7d1683a]
[2025-06-27 13:39:44,114: INFO/ForkPoolWorker-48] started running the task!
[2025-06-27 13:39:44,115: INFO/ForkPoolWorker-48] Text Alignment[4366a2d8-1385-40f1-abd2-410ed7d1683a]: {'OCR Model': 0}
[2025-06-27 13:39:44,988: INFO/ForkPoolWorker-48] Text Alignment[4366a2d8-1385-40f1-abd2-410ed7d1683a]: processing image...
[2025-06-27 13:39:44,988: WARNING/ForkPoolWorker-48] identifying text lines...
WARNING 2025-06-27 13:41:45,292 celery.redirected: /usr/local/lib/python3.7/dist-packages/tensorflow_addons/utils/ensure_tf_install.py:67: UserWarning: Tensorflow Addons supports using Python ops for all Tensorflow versions above or equal to 2.9.0 and strictly below 2.12.0 (nightly versions are not supported).
The versions of TensorFlow you are currently using is 2.5.1 and is not supported.
Some things might work, some things might not.
If you were to encounter a bug, do not file an issue.
If you want to make sure you're using a tested and supported configuration, either change the TensorFlow version or the TensorFlow Addons's version.
You can find the compatibility matrix in TensorFlow Addon's readme:
https://github.com/tensorflow/addons
UserWarning,
INFO 2025-06-27 13:41:45,424 tfaip.device.device_config: Setting up device config DeviceConfigParams(gpus=None, gpu_auto_tune=False, gpu_memory=None, soft_device_placement=True, dist_strategy=)
INFO 2025-06-27 13:41:45,502 calamari_ocr.ocr.savedmodel.sa: Checkpoint version 5 is up-to-date.
INFO 2025-06-27 13:41:45,535 tfaip.device.device_config: Setting up device config DeviceConfigParams(gpus=None, gpu_auto_tune=False, gpu_memory=None, soft_device_placement=True, dist_strategy=)
WARNING 2025-06-27 13:41:50,355 celery.redirected: Prediction: 0%| | 0/10 [00:00)
```

and
```
WARNING 2025-06-27 13:41:45,292 celery.redirected: /usr/local/lib/python3.7/dist-packages/tensorflow_addons/utils/ensure_tf_install.py:67: UserWarning: Tensorflow Addons supports using Python ops for all Tensorflow versions above or equal to 2.9.0 and strictly below 2.12.0 (nightly versions are not supported).
The versions of TensorFlow you are currently using is 2.5.1 and is not supported.
Some things might work, some things might not.
```

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.