StevenGrove / LearnableTreeFilterV2

Apache License 2.0
91 stars 9 forks source link

ImportError: /data/bbbbb/LearnableTreeFilterV2v/cvpods/_C.cpython-39-x86_64-linux-gnu.so: undefined symbol: _Z39tree_filter_refine_backward_edge_weightRKN2at6TensorES2_S2_S2_S2_S2_S2_S2_S2_ #7

Open fengenxiang1123 opened 2 years ago

fengenxiang1123 commented 2 years ago

When I execute "from cvpods.layers import TreeFilterV2", there is a problem. The detailed error is as follows: Traceback (most recent call last): File "", line 1, in File "/data/bbbbb/LearnableTreeFilterV2v/cvpods/init.py", line 3, in from .utils import setup_environment File "/data/bbbbb/LearnableTreeFilterV2v/cvpods/utils/init.py", line 27, in from .visualizer import ColorMode, VideoVisualizer, VisImage, Visualizer, colormap, random_color File "/data/bbbbb/LearnableTreeFilterV2v/cvpods/utils/visualizer/init.py", line 5, in from .video_visualizer import * File "/data/bbbbb/LearnableTreeFilterV2v/cvpods/utils/visualizer/video_visualizer.py", line 6, in from .visualizer import ColorMode, Visualizer, _create_text_labels, _PanopticPrediction File "/data/bbbbb/LearnableTreeFilterV2v/cvpods/utils/visualizer/visualizer.py", line 17, in from cvpods.structures import BitMasks, Boxes, BoxMode, Keypoints, PolygonMasks, RotatedBoxes File "/data/bbbbb/LearnableTreeFilterV2v/cvpods/structures/init.py", line 2, in from .boxes import Boxes, BoxMode, pairwise_ioa, pairwise_iou File "/data/bbbbb/LearnableTreeFilterV2v/cvpods/structures/boxes.py", line 11, in from cvpods.layers import cat File "/data/bbbbb/LearnableTreeFilterV2v/cvpods/layers/init.py", line 4, in from .deform_conv import DeformConv, ModulatedDeformConv File "/data/bbbbb/LearnableTreeFilterV2v/cvpods/layers/deform_conv.py", line 11, in from cvpods import _C ImportError: /data/bbbbb/LearnableTreeFilterV2v/cvpods/_C.cpython-39-x86_64-linux-gnu.so: undefined symbol: _Z39tree_filter_refine_backward_edge_weightRKN2at6TensorES2_S2_S2_S2_S2_S2_S2S2

StevenGrove commented 2 years ago

It seems the cuda codes of tree filter has not been compiled successfully. It could be using different cuda version for pytorch and nvcc. You can recompile the code by runing "python3 setup.py build develop --user" and check if it has similar issues.