AI Model Efficiency Toolkit (AIMET) Forum

Aimet modules not found when using docker

EDIT: Solved - After going through the Dockerfile and the script, I found that the docker is just setting up the os and python environment and aimet is not being built. aimet package has to built once I am inside the docker and paths LD_LIBRARY_PATH and PYTHONPATH has to appropriately set.

I created the docker image using the instructions on the Usage and Installation page. I also tried the script buildntest.sh. But when I tried to import aimet_common and other aimet modules in python from inside the docker, I get module not found error. Please see the attached screenshot. I haven’t changed the Dockerfile or the script.