Change Tensorflow Version in Docker-compose

Hello, I am currently setting up a new RasaX instance using the Docker-compose installation guide however I have run into an issue. During testing with the local install for Rasa I kept getting the Core Dump error when running the training, this was fixed by building Tensorflow from source for my machine. After I installed and set up the docker-compose instance I was again unable to train the model in RasaX web UI and when going into the docker container and running rasa train I was again greeted with the Core Dump error. I was able to move the built Tensorflow file into the container (rasa_rasa-x_1) however when running pip3 install --upgrade tensorflow-2.3.2-cp36-cp36m-linux_x86_64.whl im given “ERROR: tensorflow-2.3.2-cp36-cp36m-linux_x86_64.whl is not a supported wheel on this platform.”

The only thing changed after fresh install was the password.

Python version 3.8.5 (needs to be 3.6.9) pip version 21.1.2

Any help is appreciated and ill try and provide any details