ImportError: No module named '_prroi_pooling' #13

adithya-samavedhi · 2022-08-25T17:22:36Z

Hi,
I am trying to setup the codebase and run the cells part of dissect_classifier_experiment.ipynb
However, I am running into the following error: ImportError: No module named '_prroi_pooling'

Please find the Stack Trace below:

---------------------------------------------------------------------------
ImportError                               Traceback (most recent call last)
~/dissect-master/netdissect/upsegmodel/prroi_pool/functional.py in <module>
     21         [pjoin(root_dir, 'prroi_pooling_gpu.c'), pjoin(root_dir, 'prroi_pooling_gpu_impl.cu')],
---> 22         verbose=False
     23     )

/datasets/home/12/112/asamavedhi/anaconda3/envs/dissect_env2/lib/python3.6/site-packages/torch/utils/cpp_extension.py in load(name, sources, extra_cflags, extra_cuda_cflags, extra_ldflags, extra_include_paths, build_directory, verbose, with_cuda, is_python_module)
    920         with_cuda,
--> 921         is_python_module)
    922 

/datasets/home/12/112/asamavedhi/anaconda3/envs/dissect_env2/lib/python3.6/site-packages/torch/utils/cpp_extension.py in _jit_compile(name, sources, extra_cflags, extra_cuda_cflags, extra_ldflags, extra_include_paths, build_directory, verbose, with_cuda, is_python_module)
   1119         print('Loading extension module {}...'.format(name))
-> 1120     return _import_module_from_library(name, build_directory, is_python_module)
   1121 

/datasets/home/12/112/asamavedhi/anaconda3/envs/dissect_env2/lib/python3.6/site-packages/torch/utils/cpp_extension.py in _import_module_from_library(module_name, path, is_python_module)
   1440     # https://stackoverflow.com/questions/67631/how-to-import-a-module-given-the-full-path
-> 1441     file, path, description = imp.find_module(module_name, [path])
   1442     # Close the .so file after load.

/datasets/home/12/112/asamavedhi/anaconda3/envs/dissect_env2/lib/python3.6/imp.py in find_module(name, path)
    296     else:
--> 297         raise ImportError(_ERR_MSG.format(name), name=name)
    298 

ImportError: No module named '_prroi_pooling'

During handling of the above exception, another exception occurred:

ImportError                               Traceback (most recent call last)
<ipython-input-17-5fd935b128e0> in <module>
      6 classlabels = [r.split(' ')[0][3:] for r in urlopen(synset_url).read().decode('utf-8').split('\n')]
      7 classlabels = dataset.classes
----> 8 segmodel, seglabels, segcatlabels = experiment.setting.load_segmenter(args.seg)
      9 renorm = renormalize.renormalizer(dataset, target='zc')

~/dissect-master/experiment/setting.py in load_segmenter(segmenter_name)
     78     segmodels.append(segmenter.UnifiedParsingSegmenter(segsizes=[256],
     79             all_parts=all_parts,
---> 80             segdiv=('quad' if quad_seg else None)))
     81     if textures:
     82         segmenter.ensure_segmenter_downloaded('datasets/segmodel', 'texture')

~/dissect-master/netdissect/segmenter.py in __init__(self, segsizes, segdiv, all_parts)
    141         ensure_segmenter_downloaded('datasets/segmodel', 'upp')
    142         segmodel = load_unified_parsing_segmentation_model(
--> 143             segarch, segvocab, epoch)
    144         segmodel.cuda()
    145         self.segmodel = segmodel

~/dissect-master/netdissect/segmenter.py in load_unified_parsing_segmentation_model(segmodel_arch, segvocab, epoch)
    581         fc_dim=2048, use_softmax=True,
    582         nr_classes=nr_classes,
--> 583         weights=os.path.join(segmodel_dir, 'decoder_epoch_%d.pth' % epoch))
    584     segmodel = upsegmodel.SegmentationModule(
    585         seg_encoder, seg_decoder, labeldata)

~/dissect-master/netdissect/upsegmodel/models.py in build_decoder(self, nr_classes, arch, fc_dim, weights, use_softmax)
    199                 fc_dim=fc_dim,
    200                 use_softmax=use_softmax,
--> 201                 fpn_dim=512)
    202         else:
    203             raise Exception('Architecture undefined!')

~/dissect-master/netdissect/upsegmodel/models.py in __init__(self, nr_classes, fc_dim, use_softmax, pool_scales, fpn_inplanes, fpn_dim)
    255                  fpn_inplanes=(256,512,1024,2048), fpn_dim=256):
    256         # Lazy import so that compilation isn't needed if not being used.
--> 257         from .prroi_pool import PrRoIPool2D
    258         super(UPerNet, self).__init__()
    259         self.use_softmax = use_softmax

~/dissect-master/netdissect/upsegmodel/prroi_pool/__init__.py in <module>
     10 # Copyright (c) 2017 Megvii Technology Limited.
     11 
---> 12 from .prroi_pool import *
     13 

~/dissect-master/netdissect/upsegmodel/prroi_pool/prroi_pool.py in <module>
     12 import torch.nn as nn
     13 
---> 14 from .functional import prroi_pool2d
     15 
     16 __all__ = ['PrRoIPool2D']

~/dissect-master/netdissect/upsegmodel/prroi_pool/functional.py in <module>
     23     )
     24 except ImportError:
---> 25     raise ImportError('Can not compile Precise RoI Pooling library.')
     26 
     27 __all__ = ['prroi_pool2d']

ImportError: Can not compile Precise RoI Pooling library.

Could you please help resolve it?

The text was updated successfully, but these errors were encountered:

adithya-samavedhi · 2022-08-25T17:27:15Z

Here is the output for conda list:

`(dissect_env2) asamavedhi@dsmlp-jupyter-asamavedhi:~/dissect-master$ conda list

packages in environment at /datasets/home/12/112/asamavedhi/anaconda3/envs/dissect_env2:

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

ImportError: No module named '_prroi_pooling' #13

ImportError: No module named '_prroi_pooling' #13

adithya-samavedhi commented Aug 25, 2022

adithya-samavedhi commented Aug 25, 2022

ImportError: No module named '_prroi_pooling' #13

ImportError: No module named '_prroi_pooling' #13

Comments

adithya-samavedhi commented Aug 25, 2022

adithya-samavedhi commented Aug 25, 2022

packages in environment at /datasets/home/12/112/asamavedhi/anaconda3/envs/dissect_env2:

Name Version Build Channel