After some research it looks like TF version 2.4 is M1 compatible. Could you try manually changing the version in requirements.txt from tensorflow==2.3.1 to tensorflow==2.4.1 and let me know if it works? This for the conda environment method since to update the library within the Docker image the setup is more complicated. If you already have the virtual env created you can just activate it and run pip install tensorflow==2.4.1.
1 Like