NVlabs / PoseCNN-PyTorch

PyTorch implementation of the PoseCNN framework
Other
195 stars 49 forks source link

Please help - Error after running python setup.py install in lib/layers #3

Closed hrishikeshkale4 closed 3 years ago

hrishikeshkale4 commented 3 years ago

Hi, I am getting the following error after running python setup.py install in lib/layers MicrosoftTeams-image

Could you please let me know how can this be solved ?

After searching I found that a possible solution is to comment a line in /usr/local/cuda/include/crt/common_functions.h as given here https://github.com/BVLC/caffe/issues/5994#issuecomment-337507479

However, this will require admin permissions which I don't have since I am working on a remote cluster.

Any help in this regard is highly appreciated

Thanks and Regards, Hrishikesh Kale

yuxng commented 3 years ago

If you do not have sudo permission, you can use an Anaconda environment or use docker.

hrishikeshkale4 commented 3 years ago

Thanks for your response. I am using an anaconda environment. Are you suggesting that I should install cuda in the anaconda environment ? I am not sure how to achieve this

yuxng commented 3 years ago

For CUDA, you have to use a docker. I am also not sure what caused the problem you post.

hrishikeshkale4 commented 3 years ago

I am still not being able to find out what caused the problem. When I comment out the modules - 'sdf_matching_loss_kernel.cu', and 'hough_voting_kernel.cu' in th list 'sources' in the setup.py file in lib/layers, the python setup.py command works. However, when I run the demo using ./experiments/scripts/demo.sh I get the following error when importing the posecnn_cuda module

Screenshot from 2021-01-28 19-36-13

Could you please help me out with what might be causing these problems ?

Any help in this regard is highly appreciated !

Thanks Hrishikesh

dandmetal commented 3 years ago

Had the same problem. Fixed it by installing CUDA from Nvidia site: https://developer.nvidia.com/Cuda-downloads.

After installing, go to ~/.bashrc and the add lines:

export CUDA_HOME=/usr/local/cuda export LD_LIBRARY_PATH=${CUDA_HOME}/lib:$LD_LIBRARY_PATH export PATH=${CUDA_HOME}/bin:${PATH}