warp-ctc
warp-ctc copied to clipboard
SeanNaren
Whether cuda10,pytorch 1.0 is supported
root@localhost:/home/ocrtrain/train/ocr/warp-ctc/pytorch_binding# python setup.py install running install running bdist_egg running egg_info creating warpctc_pytorch.egg-info writing warpctc_pytorch.egg-info/PKG-INFO writing dependency_links to warpctc_pytorch.egg-info/dependency_links.txt writing top-level names to warpctc_pytorch.egg-info/top_level.txt writing manifest file 'warpctc_pytorch.egg-info/SOURCES.txt' reading manifest file 'warpctc_pytorch.egg-info/SOURCES.txt' writing manifest file 'warpctc_pytorch.egg-info/SOURCES.txt' installing library code to build/bdist.linux-x86_64/egg running install_lib running build_py creating build creating build/lib.linux-x86_64-3.6 creating build/lib.linux-x86_64-3.6/warpctc_pytorch copying warpctc_pytorch/init.py -> build/lib.linux-x86_64-3.6/warpctc_pytorch running build_ext building 'warpctc_pytorch._warp_ctc' extension creating build/temp.linux-x86_64-3.6 creating build/temp.linux-x86_64-3.6/src x86_64-linux-gnu-gcc -pthread -DNDEBUG -g -fwrapv -O2 -Wall -g -fstack-protector-strong -Wformat -Werror=format-security -Wdate-time -D_FORTIFY_SOURCE=2 -fPIC -I/home/michael/ocrtrain/train/ocr/warp-ctc/include -I/usr/local/lib/python3.6/dist-packages/torch/include -I/usr/local/lib/python3.6/dist-packages/torch/include/torch/csrc/api/include -I/usr/local/lib/python3.6/dist-packages/torch/include/TH -I/usr/local/lib/python3.6/dist-packages/torch/include/THC -I/usr/local/cuda/include -I/usr/include/python3.6m -c src/binding.cpp -o build/temp.linux-x86_64-3.6/src/binding.o -std=c++11 -fPIC -DWARPCTC_ENABLE_GPU -DTORCH_API_INCLUDE_EXTENSION_H -DTORCH_EXTENSION_NAME=_warp_ctc -D_GLIBCXX_USE_CXX11_ABI=0 src/binding.cpp:10:11: fatal error: ATen/cuda/CUDAGuard.h: No such file or directory #include "ATen/cuda/CUDAGuard.h" ^~~~~~~~~~~~~~~~~~~~~~~ compilation terminated. error: command 'x86_64-linux-gnu-gcc' failed with exit status 1
I have a fata error when i execute "python setup.py install", couldn't find "ATen/cuda/CUDAGuard.h", how to fix it.
PyTorch 1.0.1 support CTCLoss itself. Use
torch.nn.CTCLoss
warp-ctc支持window吗?
You can use pip install warpctc-pytorch10-cuda90==0.1.3 instead of "python setup.py install, note the right version of warpctc-pytorch10-cuda90
