diff --git a/Dockerfile b/Dockerfile
index f27ea6074c2a6dba59d376584339cbcf29c4c8d0..f64e64f75bb4390f9336c2d85a6ebcd8ce3cc682 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,7 +1,6 @@
-FROM yoanlin/opencv-python3:latest
+FROM tensorflow/tensorflow:2.0.0a0-gpu-py3
 RUN apt-get update
 RUN pip install --upgrade pip
-RUN pip install tensorflow==2.0.0-beta1 
 RUN pip install numpy scipy scikit-learn pillow h5py keras 
 RUN pip install --upgrade imutils 
 RUN pip install --upgrade scikit-learn