Course Wiki

Preparing a Docker image — Machine Learning Zoomcamp Module 9

Refer to updates.md for info on running TF lite in 2024.

Machine Learning Zoomcamp › Module 9: Serverless Deep Learning › Preparing a Docker image

Notes

Refer to updates.md for info on running TF lite in 2024.

Using pip install for TF-Lite binaries

When using pip to install the compiled binary, make sure you use the raw file, not a link to the github page.

Correct:

pip install https://github.com/alexeygrigorev/tflite-aws-lambda/raw/main/tflite/tflite_runtime-2.14.0-cp310-cp310-linux_x86_64.whl

(Note /raw/ in the path)

Also correct:

pip install https://github.com/alexeygrigorev/tflite-aws-lambda/blob/main/tflite/tflite_runtime-2.14.0-cp310-cp310-linux_x86_64.whl?raw=true

The wheel file above is for Python 3.10. Check other available compiled TF lite versions here.

Not correct - won’t work:

pip install https://github.com/alexeygrigorev/tflite-aws-lambda/blob/main/tflite/tflite_runtime-2.14.0-cp310-cp310-linux_x86_64.whl

If the file is incorrect, you’ll get an error message like that:

zipfile.BadZipFile: File is not a zip file

ENTRYPOINT vs CMD

This link explains the difference between them: https://stackoverflow.com/a/34245657

ENTRYPOINT specifies a command that will always be executed when the container starts. CMD specifies arguments that will be fed to the ENTRYPOINT.

In case of the lambda base pacakge, the authors already specified the entrypoint and we only need to overwrite the arguments passed to the entrypoint,

⚠️ The notes are written by the community.
If you see an error here, please create a PR with a fix.

Key concepts

Sources


DataTalks.Club. Hosted on GitHub Pages. Built with Rustkyll. We use cookies.