search

mks0601 / TF-SimpleHumanPose

TensorFlow implementation of "Simple Baselines for Human Pose Estimation and Tracking", ECCV 2018
336 stars 90 forks source link

__cudaRegisterFatBinaryEnd #44

Closed IshrakHasin closed 4 years ago

IshrakHasin commented 5 years ago

I have not changed anything in the code and using the mentioned configurations described in the project. Finished the training. I am getting an error. Does anyone have any idea? The traceback is: Traceback (most recent call last): File "test.py", line 24, in from nms.nms import oks_nms File "/datadrive/common/Ishrak/Integral_Pose_estimation/TF-SimpleHumanPose/main/../lib/nms/nms.py", line 14, in from .gpu_nms import gpu_nms ImportError: /datadrive/common/Ishrak/Integral_Pose_estimation/TF-SimpleHumanPose/main/../lib/nms/gpu_nms.cpython-36m-x86_64-linux-gnu.so: undefined symbol: __cudaRegisterFatBinaryEnd

Best Ishrak Technos Data Science Engineering Inc, Tokyo

mks0601 commented 5 years ago

Did you do this? Run cd ${POSE_ROOT}/lib and make to build NMS modules. This is from README.

IshrakHasin commented 5 years ago

Yes, I did. Still, the problem persists.

mks0601 commented 5 years ago

I think that is something related with crush between NMS build version and your TF version. Could you re-build NMS?