Create a docker image

[chongma]

It would be easier to use if there is a docker configuration. I am trying to set it up on Fedora 34 and it has python 3.9 and it is quite hard to get everything set up as suggested in the readme. Providing a docker configuration would allow an easy to use compatibility reference. People could configure their own volumes where the code is stored. could be based on tensorflow gpu dockerfile?