No module named 'warpctc_pytorch'

1.

Traceback (most recent call last):
  File "/home/###/Downloads/softmare/pycharm-community-2018.3.2/helpers/pydev/pydevd.py", line 1741, in <module>
    main()
  File "/home/###/Downloads/softmare/pycharm-community-2018.3.2/helpers/pydev/pydevd.py", line 1735, in main
    globals = debugger.run(setup['file'], None, None, is_module)
  File "/home/###/Downloads/softmare/pycharm-community-2018.3.2/helpers/pydev/pydevd.py", line 1135, in run
    pydev_imports.execfile(file, globals, locals)  # execute the script
  File "/home/###/Downloads/softmare/pycharm-community-2018.3.2/helpers/pydev/_pydev_imps/_pydev_execfile.py", line 18, in execfile
    exec(compile(contents+"\n", file, 'exec'), glob, loc)
  File "/home/###/sdb/CODE/ocr/model/crnn識別/train.py", line 12, in <module>
    from warpctc_pytorch import CTCLoss
ModuleNotFoundError: No module named 'warpctc_pytorch'

報錯原因:CTC+pytorch編譯配置warp-CTC;

warp-CTC 安裝:

git clone https://github.com/SeanNaren/warp-ctc.git
cd warp-ctc
mkdir build
cd build
cmake ..
make
cd ../pytorch_binding
python setup.py install

安裝過程中遇到的問題:

報錯:

c -D_GLIBCXX_USE_CXX11_ABI=0
cc1plus: 警告:command line option ‘-Wstrict-prototypes’ is valid for C/ObjC but not for C++
src/binding.cpp:10:33: 致命錯誤:c10/cuda/CUDAGuard.h:沒有那個文件或目錄

解決方法:

刪除 pytorch-binding/src/binding.cpp 中的 #include <c10/cuda/CUDAGuard.h>即可

參考鏈接:

1.安裝參考:https://blog.csdn.net/zzc15806/article/details/83999668

2.報錯參考:https://blog.csdn.net/amu76980/article/details/101173862

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章