diff --git a/easy_rec/__init__.py b/easy_rec/__init__.py
index 26e3c753f..08f6fc377 100644
--- a/easy_rec/__init__.py
+++ b/easy_rec/__init__.py
@@ -16,13 +16,16 @@
 
 logging.basicConfig(
     level=logging.INFO, format='[%(asctime)s][%(levelname)s] %(message)s')
+
+if tf.__version__ >= '2.0':
+  tf = tf.compat.v1
+
 tf.logging.set_verbosity(tf.logging.INFO)
 
 # Avoid import tensorflow which conflicts with the version used in EasyRecProcessor
 if 'PROCESSOR_TEST' not in os.environ:
   if platform.system() == 'Linux':
     ops_dir = os.path.join(curr_dir, 'python/ops')
-    import tensorflow as tf
     if 'PAI' in tf.__version__:
       ops_dir = os.path.join(ops_dir, '1.12_pai')
     elif tf.__version__.startswith('1.12'):
diff --git a/easy_rec/version.py b/easy_rec/version.py
index 235c9c2a6..bcce3f37f 100644
--- a/easy_rec/version.py
+++ b/easy_rec/version.py
@@ -1,3 +1,3 @@
 # -*- encoding:utf-8 -*-
 # Copyright (c) Alibaba, Inc. and its affiliates.
-__version__ = '0.7.6'
+__version__ = '0.7.7'