-
Notifications
You must be signed in to change notification settings - Fork 282
Closed
Description
Environment: Docker
OS: ubuntu 16.04 LTS 64-bit
Processor: Intel(R) Core i7-7700 CPU @ 3.60 GHz * 8
Tensorflow version: 2.6.0
TF_ENABLE_ONEDNN_OPTS=1
Hi, I am currently trying to run Tensorflow 2.6 quantization with Neural Compressor in Docker environment. Basically I modified my image from tensorflow/tensorflow:2.6.0-gpu-jupyter, and use pip install neural-compressor
to install. However when I try to import neural_compressor with the command
$ python
>>> import neural_compressor
the following error occurred:
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
File "/usr/local/lib/python3.6/dist-packages/neural_compressor/__init__.py", line 18, in <module>
from .quantization import Quantization
File "/usr/local/lib/python3.6/dist-packages/neural_compressor/quantization.py", line 20, in <module>
from .data import DATALOADERS, DATASETS
File "/usr/local/lib/python3.6/dist-packages/neural_compressor/data/__init__.py", line 19, in <module>
from .dataloaders import DataLoader
File "/usr/local/lib/python3.6/dist-packages/neural_compressor/data/dataloaders/__init__.py", line 18, in <module>
from .dataloader import DataLoader
File "/usr/local/lib/python3.6/dist-packages/neural_compressor/data/dataloaders/dataloader.py", line 18, in <module>
from neural_compressor.experimental.data.dataloaders import DATALOADERS
File "/usr/local/lib/python3.6/dist-packages/neural_compressor/experimental/__init__.py", line 18, in <module>
from .component import Component
File "/usr/local/lib/python3.6/dist-packages/neural_compressor/experimental/component.py", line 17, in <module>
from ..conf.config import Conf
File "/usr/local/lib/python3.6/dist-packages/neural_compressor/conf/config.py", line 21, in <module>
from ..strategy import STRATEGIES
File "/usr/local/lib/python3.6/dist-packages/neural_compressor/strategy/__init__.py", line 18, in <module>
from .strategy import STRATEGIES
File "/usr/local/lib/python3.6/dist-packages/neural_compressor/strategy/strategy.py", line 30, in <module>
from ..utils.create_obj_from_config import create_eval_func, create_train_func
File "/usr/local/lib/python3.6/dist-packages/neural_compressor/utils/create_obj_from_config.py", line 19, in <module>
from neural_compressor.experimental.data import DATASETS, TRANSFORMS, FILTERS, DATALOADERS
File "/usr/local/lib/python3.6/dist-packages/neural_compressor/experimental/data/__init__.py", line 20, in <module>
from .transforms import TRANSFORMS, BaseTransform, transform_registry
File "/usr/local/lib/python3.6/dist-packages/neural_compressor/experimental/data/transforms/__init__.py", line 18, in <module>
from .transform import TRANSFORMS, BaseTransform, transform_registry
File "/usr/local/lib/python3.6/dist-packages/neural_compressor/experimental/data/transforms/transform.py", line 316, in <module>
'nearest': cv2.INTER_NEAREST,
File "/usr/local/lib/python3.6/dist-packages/neural_compressor/utils/utility.py", line 72, in __getattr__
top_level_module = __import__(self.module_name)
File "/usr/local/lib/python3.6/dist-packages/cv2/__init__.py", line 180, in <module>
bootstrap()
File "/usr/local/lib/python3.6/dist-packages/cv2/__init__.py", line 152, in bootstrap
native_module = importlib.import_module("cv2")
File "/usr/lib/python3.6/importlib/__init__.py", line 126, in import_module
return _bootstrap._gcd_import(name[level:], package, level)
ImportError: libGL.so.1: cannot open shared object file: No such file or directory
I have attached my Dockerfile, (please note that neural_compressor is installed after creating the image and the container is committed successfully)
Dockerfile.txt
Thanks!
Metadata
Metadata
Assignees
Labels
No labels