search

SeanNaren / warp-ctc

Pytorch Bindings for warp-ctc
Apache License 2.0
756 stars 271 forks source link

Whether cuda10,pytorch 1.0 is supported #121

Open Joker19920330 opened 5 years ago

Joker19920330 commented 5 years ago

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.

Holmeyoung commented 5 years ago

PyTorch 1.0.1 support CTCLoss itself. Use torch.nn.CTCLoss

Joker19920330 commented 5 years ago

@Holmeyoung Thank you very much

xiiiiiiii commented 5 years ago

PyTorch 1.0.1 support CTCLoss itself. Use torch.nn.CTCLoss

warp-ctc支持window吗?

shuaijiang commented 4 years ago

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