search

fh2019ustc / DocTr

The official code for “DocTr: Document Image Transformer for Geometric Unwarping and Illumination Correction”, ACM MM, Oral Paper, 2021.
Other
345 stars 48 forks source link

AssertionError: Torch not compiled with CUDA enabled #25

Open 15991361037 opened 1 year ago

15991361037 commented 1 year ago

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