I have installed the following docker image with the jupyter notebook installed - https://hub.docker.com/layers/tensorflow/tensorflow/2.0.4-gpu-jupyter/images/sha256-926eb250274