AssertionError: Torch not compiled with CUDA enabled

[15991361037]

Traceback (most recent call last):
File "E:\jiaozheng\DocTr-main\inference.py", line 138, in
main()
File "E:\jiaozheng\DocTr-main\inference.py", line 134, in main
rec(opt)
File "E:\jiaozheng\DocTr-main\inference.py", line 76, in rec
GeoTr_Seg_model = GeoTr_Seg().cuda()
File "E:\jiaozheng\DocTr-main\venv\lib\site-packages\torch\nn\modules\module.py", line 905, in cuda
return self._apply(lambda t: t.cuda(device))
File "E:\jiaozheng\DocTr-main\venv\lib\site-packages\torch\nn\modules\module.py", line 797, in _apply
module._apply(fn)
File "E:\jiaozheng\DocTr-main\venv\lib\site-packages\torch\nn\modules\module.py", line 797, in _apply
module._apply(fn)
File "E:\jiaozheng\DocTr-main\venv\lib\site-packages\torch\nn\modules\module.py", line 797, in _apply
module._apply(fn)
[Previous line repeated 1 more time]
File "E:\jiaozheng\DocTr-main\venv\lib\site-packages\torch\nn\modules\module.py", line 820, in _apply
param_applied = fn(param)
File "E:\jiaozheng\DocTr-main\venv\lib\site-packages\torch\nn\modules\module.py", line 905, in
return self.apply(lambda t: t.cuda(device))
File "E:\jiaozheng\DocTr-main\venv\lib\site-packages\torch\cuda_init.py", line 239, in _lazy_init
raise AssertionError("Torch not compiled with CUDA enabled")
AssertionError: Torch not compiled with CUDA enabled

[Tarek-Eladly]

Open the project in any code editor => replace all in all the files cuda with cpu ... performance will drop

[fh2019ustc]

@Tarek-Eladly Ah... I am also confused.

[Tarek-Eladly]

Some devices/setups don't support GPU(CUDA), so switching to CPU to make it work ... you need to configure/setup your system to support GPU(CUDA) so it works correctly

[Tarek-Eladly]

@fh2019ustc Thank you for this wonderful Doc preparation tool for OCR really thanks

[fh2019ustc]

@Tarek-Eladly Glad it's helpful for you! Feel free to let me know if you encounter any other issues. 😊