Make sure dockerfile supports both amd64 and arm64

Description

Make sure dockerfile supports both amd64 and arm64.

Related to #69 (closed)

Reviewers list

• @marcin.chron

Developer Self Check:

Strike through any lines that are not applicable (~~line~~) then check the box

• Code is clear and sufficiently documented (classes, methods and function docs, document important assumptions, link to reference materials)
• Type annotations
• Issue is in the review column
• Code has been pre-reviewed by developer
• Sensitive elements have been highlighted with opened discussions