The google colab moved to python 3.7 but aimet modules are compiled with python 3.6. This is problematic:
please see: /usr/local/lib/python3.7/dist-packages/aimet_torch/qc_quantize_op.py in 48 from aimet_torch.tensor_quantizer import PostTrainingTensorQuantizer 49 import aimet_torch.quantsim_straight_through_grad as ste —> 50 import libpymo 51 52
ImportError: Python version mismatch: module was compiled for Python 3.6, but the interpreter version is incompatible: 3.7.10 (default, Feb 20 2021, 21:17:23) [GCC 7.5.0].
We are working on an AIMET release with python 3.7 - specifically for supporting Google Colab.
Meanwhile - if you are able to recompile AIMET, please do so, so you are not blocked.
Hi @sarakaram, apologies for the delay. Our projected date to have the Python 3.7 version of AIMET package available for Code Collab is May 3. We’ll also update the instructions accordingly.