diff --git a/Dockerfile b/Dockerfile
index f89aa2069e8e3d4d3f991b3f33dae59620b8752a..8e4e7a636c0b04be8538cb54607e74d768fc279d 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -15,7 +15,7 @@
 
 # A dev image based on paddle production image
 
-FROM paddlepaddle/paddle:latest-gpu
+FROM paddlepaddle/paddle:1.1.0-gpu-cuda9.0-cudnn7
 
 RUN apt-get install -y cmake
 RUN pip install -i https://pypi.tuna.tsinghua.edu.cn/simple gym
diff --git a/parl/layers/layer_wrappers.py b/parl/layers/layer_wrappers.py
index db293a65912078082dfc07d7c88f81ef465b3a97..fcafe61f9467ffb57e83b90a83475ae80ac18f8b 100644
--- a/parl/layers/layer_wrappers.py
+++ b/parl/layers/layer_wrappers.py
@@ -179,7 +179,6 @@ def fc(size,
        num_flatten_dims=1,
        param_attr=None,
        bias_attr=None,
-       use_mkldnn=False,
        act=None,
        name=None):
     """
@@ -200,7 +199,6 @@ def fc(size,
                              num_flatten_dims=num_flatten_dims,
                              param_attr=self.param_attr,
                              bias_attr=self.bias_attr,
-                             use_mkldnn=use_mkldnn,
                              act=act,
                              is_test=is_test)
 
@@ -410,7 +408,6 @@ def conv2d(num_filters,
            param_attr=None,
            bias_attr=None,
            use_cudnn=True,
-           use_mkldnn=False,
            act=None,
            name=None):
     """
@@ -437,7 +434,6 @@ def conv2d(num_filters,
                 param_attr=self.param_attr,
                 bias_attr=self.bias_attr,
                 use_cudnn=use_cudnn,
-                use_mkldnn=use_mkldnn,
                 act=act)
 
     return Conv2D_()