search

nv-tlabs / GSCNN

Gated-Shape CNN for Semantic Segmentation (ICCV 2019)
https://nv-tlabs.github.io/GSCNN/
Other
915 stars 200 forks source link

ImportError: No module named 'enclib_cpu' #85

Open yyt111222 opened 2 years ago

yyt111222 commented 2 years ago

Hi, I met a package bug when trying to evaluate the method by 'python train.py --evaluate --snapshot checkpoints/best_cityscapes_checkpoint.pth'

The bug is : Traceback (most recent call last): File "train.py", line 380, in main() File "train.py", line 127, in main assert_and_infer_cfg(args) File "C:\Users\asd\Desktop\GSCNN-master\config.py", line 86, in assert_and_infer_cfg import encoding File "D:\anconda3\envs\GSCNN0415\lib\site-packages\encoding__init.py", line 13, in from . import nn, functions, parallel, utils, models, datasets, transforms File "D:\anconda3\envs\GSCNN0415\lib\site-packages\encoding\nn\init.py", line 12, in from .encoding import * File "D:\anconda3\envs\GSCNN0415\lib\site-packages\encoding\nn\encoding.py", line 18, in from ..functions import scaled_l2, aggregate, pairwise_cosine File "D:\anconda3\envs\GSCNN0415\lib\site-packages\encoding\functions\init.py", line 2, in from .encoding import * File "D:\anconda3\envs\GSCNN0415\lib\site-packages\encoding\functions\encoding.py", line 14, in from .. import lib File "D:\anconda3\envs\GSCNN0415\lib\site-packages\encoding\lib\init__.py", line 15, in ], build_directory=cpu_path, verbose=False) File "D:\anconda3\envs\GSCNN0415\lib\site-packages\torch\utils\cpp_extension.py", line 644, in load is_python_module) File "D:\anconda3\envs\GSCNN0415\lib\site-packages\torch\utils\cpp_extension.py", line 824, in _jit_compile return _import_module_from_library(name, build_directory, is_python_module) File "D:\anconda3\envs\GSCNN0415\lib\site-packages\torch\utils\cpp_extension.py", line 967, in _import_module_from_library file, path, description = imp.find_module(module_name, [path]) File "D:\anconda3\envs\GSCNN0415\lib\imp.py", line 297, in find_module raise ImportError(_ERR_MSG.format(name), name=name) ImportError: No module named 'enclib_cpu'

It seems the unmatched packages in my conda env: pytorch 1.1.0 python 3.6.13 torch-encoding 1.0.1 torchvision 0.3.0

So, is the torch encoding unmatched? Could you tell me how to fix it. Thank you!