stickcui / warp-ctc-pytorch Goto Github PK

I'm using torch 0.4.0 and python 3.6
getting this stack trace when running python3 setup.py install command:::

running install
running bdist_egg
running egg_info
creating warpctc.egg-info
writing warpctc.egg-info/PKG-INFO
writing dependency_links to warpctc.egg-info/dependency_links.txt
writing requirements to warpctc.egg-info/requires.txt
writing top-level names to warpctc.egg-info/top_level.txt
writing manifest file 'warpctc.egg-info/SOURCES.txt'
reading manifest file 'warpctc.egg-info/SOURCES.txt'
writing manifest file 'warpctc.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
copying warpctc/init.py -> build/lib.linux-x86_64-3.6/warpctc
copying warpctc/warp_ctc.py -> build/lib.linux-x86_64-3.6/warpctc
running build_ext
building 'warpctc._warp_ctc' extension
creating build/temp.linux-x86_64-3.6
creating build/temp.linux-x86_64-3.6/warpctc
creating build/temp.linux-x86_64-3.6/warpctc/src
creating build/temp.linux-x86_64-3.6/warpctc/src/cpu
creating build/temp.linux-x86_64-3.6/warpctc/src/cuda
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 -DWITH_CUDA -DPYTORCH_VER_0_4 -Iwarpctc/core/include -Iwarpctc/src -I/home/rajeev/environments/ds3/lib/python3.6/site-packages/torch/lib/include -I/home/rajeev/environments/ds3/lib/python3.6/site-packages/torch/lib/include/TH -I/home/rajeev/environments/ds3/lib/python3.6/site-packages/torch/lib/include/THC -I/usr/local/cuda/include -I/home/rajeev/environments/ds3/include -I/usr/include/python3.6m -c warpctc/src/version.cpp -o build/temp.linux-x86_64-3.6/warpctc/src/version.o -std=c++11 -fPIC -DTORCH_EXTENSION_NAME=warpctc._warp_ctc
In file included from warpctc/src/cpu/version.h:4:0,
from warpctc/src/version.cpp:4:
/home/rajeev/environments/ds3/lib/python3.6/site-packages/torch/lib/include/torch/torch.h:3:20: fatal error: Python.h: No such file or directory
compilation terminated.
error: command 'x86_64-linux-gnu-gcc' failed with exit status 1

Any Help on this ??

gcc -pthread -fno-strict-aliasing -O2 -g -pipe -Wall -Wp,-D_FORTIFY_SOURCE=2 -fexceptions -fstack-protector-strong --param=ssp-buffer-size=4 -grecord-gcc-switches -m64 -mtune=generic -D_GNU_SOURCE -fPIC -fwrapv -DNDEBUG -O2 -g -pipe -Wall -Wp,-D_FORTIFY_SOURCE=2 -fexceptions -fstack-protector-strong --param=ssp-buffer-size=4 -grecord-gcc-switches -m64 -mtune=generic -D_GNU_SOURCE -fPIC -fwrapv -fPIC -DWITH_CUDA -DPYTORCH_VER_0_4 -Iwarpctc/core/include -Iwarpctc/src -I/usr/lib64/python2.7/site-packages/torch/lib/include -I/usr/lib64/python2.7/site-packages/torch/lib/include/TH -I/usr/lib64/python2.7/site-packages/torch/lib/include/THC -I/usr/local/cuda/include -I/usr/include/python2.7 -c warpctc/src/cpu/cpu_ctc.cpp -o build/temp.linux-x86_64-2.7/warpctc/src/cpu/cpu_ctc.o -std=c++11 -fPIC -DTORCH_EXTENSION_NAME=warp_ctc -D_GLIBCXX_USE_CXX11_ABI=0
/usr/local/cuda/bin/nvcc -DWITH_CUDA -DPYTORCH_VER_0_4 -Iwarpctc/core/include -Iwarpctc/src -I/usr/lib64/python2.7/site-packages/torch/lib/include -I/usr/lib64/python2.7/site-packages/torch/lib/include/TH -I/usr/lib64/python2.7/site-packages/torch/lib/include/THC -I/usr/local/cuda/include -I/usr/include/python2.7 -c warpctc/src/cuda/gpu_ctc.cu -o build/temp.linux-x86_64-2.7/warpctc/src/cuda/gpu_ctc.o -DCUDA_HAS_FP16=1 -D__CUDA_NO_HALF_OPERATORS_ -D__CUDA_NO_HALF_CONVERSIONS__ -D__CUDA_NO_HALF2_OPERATORS__ -Wno-deprecated-gpu-targets -DTORCH_EXTENSION_NAME=_warp_ctc -D_GLIBCXX_USE_CXX11_ABI=0 --compiler-options '-fPIC' -std=c++11
unable to execute /usr/local/cuda/bin/nvcc: No such file or directory
error: command '/usr/local/cuda/bin/nvcc' failed with exit status 1
请问如何解决，谢谢