tmbdev/clstm

question: clstmocrtrain on GPU

bugsyb opened this issue · 0 comments

bugsyb commented

Great thanks for all superb work!
Trying to run clstmocrtrain on CUDA, doesnt' seem to work.
I've compiled everything with "scons -j7 gpu=1" (I've 8 cores).
No errors were produced (well after all it took me a bit of time to get here, but compilation worked).

Now running gpu=1 clstmocrtrain + rest of parameters seems to occupy CPU only and nothing is shown on GPU.

Question is how to validate if all is prepared correctly and why clstmocrtrain is not running on GPU?
Should there be a recommendation to use another GPU enabled tool to generate valid LSTM for Ocropy, I'm very open to that. I will try Kraken, but wanted to first try clstmocrtrain.

Environment:
Mint 18.2 (ubuntu 16.04),
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 384.130 Driver Version: 384.130 |
|-------------------------------+----------------------+----------------------+
| GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. |
|===============================+======================+======================|
| 0 Quadro M1200 Off | 00000000:01:00.0 Off | N/A |
| N/A 47C P8 N/A / N/A | 8MiB / 4044MiB | 0% Default |
+-------------------------------+----------------------+----------------------+

+-----------------------------------------------------------------------------+
| Processes: GPU Memory |
| GPU PID Type Process name Usage |
|=============================================================================|
| 0 10592 G /usr/lib/xorg/Xorg 7MiB |
+-----------------------------------------------------------------------------+

top - 03:04:28 up 39 min, 8 users, load average: 2,86, 3,37, 2,77
Tasks: 320 total, 3 running, 316 sleeping, 0 stopped, 1 zombie
%Cpu(s): 23,1 us, 1,5 sy, 0,0 ni, 74,7 id, 0,6 wa, 0,0 hi, 0,1 si, 0,0 st
KiB Mem : 32787312 total, 27967764 free, 1509580 used, 3309968 buff/cache
KiB Swap: 0 total, 0 free, 0 used. 29756676 avail Mem

PID USER PR NI VIRT RES SHR S %CPU %MEM TIME+ COMMAND
10642 myself 20 0 40120 13536 5696 R 100,0 0,0 23:28.40 clstmocrtrain

command:
saventrain=201 hidden=50 lrate=1e-2 save_name=gpu-test ~/OCR/clstm/clstmocrtrain -v ./manifest

Thank you in advance for help.