방금 확인해본 바로는,

tensorflow/tensorflow:2.1.0-gpu-py3

으로 설치하면 실행이 됩니다.