Probelm with PyTorch knn

Hey,

thanks for the available code for DenseFusion! I want to use it for my own synthetic dataset (created with NDDS), but I’ve got some problems getting started with DenseFusion.

My System:

• Ubuntu 18.02
• Cuda 10.2
• Python 3.6.9
• PyTorch 0.4.1
• Torchvision 0.2.2

The error: When running ./experiments/scripts/eval_linemod.sh I get the following error message:

+ set -e
+ export PYTHONUNBUFFERED=True
+ PYTHONUNBUFFERED=True
+ export CUDA_VISIBLE_DEVICES=0
+ CUDA_VISIBLE_DEVICES=0
+ python3 ./tools/eval_linemod.py --dataset_root ./datasets/linemod/Linemod_preprocessed --model trained_checkpoints/linemod/pose_model_9_0.01310166542980859.pth --refine_model trained_checkpoints/linemod/pose_refine_model_493_0.006761023565178073.pth
Traceback (most recent call last):
  File "./tools/eval_linemod.py", line 20, in <module>
    from lib.loss import Loss
  File "/home/katharina/Schreibtisch/DenseFusion/lib/loss.py", line 9, in <module>
    from lib.knn.__init__ import KNearestNeighbor
  File "/home/katharina/Schreibtisch/DenseFusion/lib/knn/__init__.py", line 7, in <module>
    from lib.knn import knn_pytorch as knn_pytorch
  File "/home/katharina/Schreibtisch/DenseFusion/lib/knn/knn_pytorch/__init__.py", line 3, in <module>
    from ._knn_pytorch import lib as _lib, ffi as _ffi
ImportError: /home/katharina/Schreibtisch/DenseFusion/lib/knn/knn_pytorch/_knn_pytorch.so: undefined symbol: state

I don’t where that error comes from. Can you help me with that issue please?

[KatharinaSchmidt]

I rebuilt the knn module like in #33 Afterwards there were some minor fixed needed (but I don’t remember them all).

[Destinycjk]

Hello! @drapado Thank you very much for your contribution! I aslo face the same problem /code/DenseFusion-Pytorch/lib/knn/knn_pytorch.cpython-36m-x86_64-linux-gnu.so: undefined symbol: _ZN6caffe26detail37_typeMetaDataInstance_preallocated_10E I follow the steps as you introduced in https://github.com/j96w/DenseFusion/pull/170. Do you know how to solve the problem? Thank you!