search

princeton-vl / pose-ae-train

Training code for "Associative Embedding: End-to-End Learning for Joint Detection and Grouping"
BSD 3-Clause "New" or "Revised" License
373 stars 76 forks source link

fatal error: TH/TH.h: No such file or directory #37

Closed JinyuJinyuJinyu closed 5 years ago

JinyuJinyuJinyu commented 5 years ago

when i run python build.py install encountered this issue in ubuntu 16.04. I studying this project, could anyone tell me how to solve this! thanks ahead.

issue

Including CUDA code. /home/xgx/anaconda3/envs/py36/lib/python3.6/distutils/extension.py:131: UserWarning: Unknown Extension options: 'headers', 'relative_to', 'with_cuda' warnings.warn(msg) running install running build running build_ext building 'my_lib' extension gcc -pthread -B /home/jinyu/anaconda3/envs/py36/compiler_compat -Wl,--sysroot=/ -Wsign-compare -DNDEBUG -g -fwrapv -O3 -Wall -Wstrict-prototypes -fPIC -DWITH_CUDA -I/home/jinyu/anaconda3/envs/py36/include/python3.6m -c src/my_lib.c -o build/temp.linux-x86_64-3.6/src/my_lib.o src/my_lib.c:1:19: fatal error: TH/TH.h: No such file or directory compilation terminated. error: command 'gcc' failed with exit status 1

py36 envrioment

conda list

Name Version Build Channel blas 1.0 mkl
bzip2 1.0.6 h14c3975_5
ca-certificates 2019.3.9 hecc5488_0 conda-forge cairo 1.14.12 h8948797_3
certifi 2019.3.9 py36_0 conda-forge cffi 1.12.2 py36h2e261b9_1
cycler 0.10.0 py_1 conda-forge cython 0.29.7 py36he1b5a44_0 conda-forge dbus 1.13.2 h714fa37_1
expat 2.2.5 hf484d3e_1002 conda-forge ffmpeg 3.4 h7985aa0_0
fontconfig 2.13.1 he4413a7_1000 conda-forge freeglut 3.0.0 hf484d3e_5
freetype 2.10.0 he983fc9_0 conda-forge glib 2.56.2 hd408876_0
graphite2 1.3.13 h23475e2_0
gst-plugins-base 1.14.0 hbbd80ab_1
gstreamer 1.14.0 hb453b48_1
h5py 2.8.0 py36h39dcb92_0
harfbuzz 1.9.0 he243708_1001 conda-forge hdf5 1.8.18 h6792536_1
icu 58.2 h9c2bf20_1
intel-openmp 2019.3 199
jasper 1.900.1 4 conda-forge jpeg 9b h024ee3a_2
kiwisolver 1.0.1 py36h6bb024c_1002 conda-forge libedit 3.1.20181209 hc058e9b_0
libffi 3.2.1 hd88cf55_4
libgcc-ng 8.2.0 hdf63c60_1
libgfortran-ng 7.3.0 hdf63c60_0
libglu 9.0.0 hf484d3e_1
libopus 1.3 h7b6447c_0
libpng 1.6.36 hbc83047_0
libprotobuf 3.5.2 h6f1eeef_0
libstdcxx-ng 8.2.0 hdf63c60_1
libtiff 4.0.10 h2733197_2
libuuid 2.32.1 h14c3975_1000 conda-forge libvpx 1.7.0 h439df22_0
libxcb 1.13 h1bed415_1
libxml2 2.9.9 he19cac6_0
matplotlib 3.0.2 py36h8a2030e_1001 conda-forge matplotlib-base 3.0.2 py36h167e16e_1001 conda-forge mkl 2019.3 199
mkl_fft 1.0.10 py36ha843d7b_0
mkl_random 1.0.2 py36hd81dba3_0
munkres 1.0.7 py36_0 omnia ncurses 6.1 he6710b0_1
ninja 1.9.0 py36hfd86e86_0
numpy 1.16.2 py36h7e9f1db_0
numpy-base 1.16.2 py36hde5b4d6_0
olefile 0.46 py36_0
opencv 3.3.1 py36h0f6f1c3_0
openssl 1.1.1b h14c3975_1 conda-forge pcre 8.43 he6710b0_0
pillow 5.4.1 py36h34e0f95_0
pip 19.0.3 py36_0
pixman 0.38.0 h7b6447c_0
pycocotools 2.0.0 py36h14c3975_1000 conda-forge pycparser 2.19 py36_0
pyparsing 2.4.0 py_0 conda-forge pyqt 5.6.0 py36h13b7fb3_1008 conda-forge python 3.6.8 h0371630_0
python-dateutil 2.8.0 py_0 conda-forge pytorch 1.0.0 py3.6_cuda9.0.176_cudnn7.4.1_1 soumith qt 5.6.3 h8bf5577_3
readline 7.0 h7b6447c_5
scipy 1.1.0 py36h7c811a0_2
setuptools 41.0.0 py36_0
sip 4.18.1 py36hf484d3e_1000 conda-forge six 1.12.0 py36_0
sqlite 3.27.2 h7b6447c_0
tk 8.6.8 hbc83047_0
torchvision 0.2.1 py_2 soumith tornado 6.0.2 py36h516909a_0 conda-forge tqdm 4.31.1 py36_1
wheel 0.33.1 py36_0
xz 5.2.4 h14c3975_4
zlib 1.2.11 h7b6447c_3
zstd 1.3.7 h0b5b093_0

cuda version

nvcc --version

nvcc: NVIDIA (R) Cuda compiler driver Copyright (c) 2005-2017 NVIDIA Corporation Built on Fri_Sep__1_21:08:03_CDT_2017 Cuda compilation tools, release 9.0, V9.0.176

variable path

gedit .bashrc

cuda things

export PATH=/usr/local/cuda-9.0/bin${PATH:+:${PATH}} export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64\${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}} export CPATH=/usr/local/cuda-9.0/include

cuda end