diff --git a/cs_gan/cs.py b/cs_gan/cs.py
index ebe171e..e59f31c 100644
--- a/cs_gan/cs.py
+++ b/cs_gan/cs.py
@@ -19,7 +19,7 @@ from __future__ import print_function
 import math
 
 import sonnet as snt
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 
 from cs_gan import utils
 
diff --git a/cs_gan/gan.py b/cs_gan/gan.py
index 6f0ec97..a50db7f 100644
--- a/cs_gan/gan.py
+++ b/cs_gan/gan.py
@@ -16,7 +16,7 @@ import collections
 import math
 
 import sonnet as snt
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 
 from cs_gan import utils
 
diff --git a/cs_gan/main.py b/cs_gan/main.py
index c4ba3be..ea46fc6 100644
--- a/cs_gan/main.py
+++ b/cs_gan/main.py
@@ -21,7 +21,7 @@ import os
 from absl import app
 from absl import flags
 from absl import logging
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 
 from cs_gan import file_utils
 from cs_gan import gan
diff --git a/cs_gan/main_cs.py b/cs_gan/main_cs.py
index 26b459d..ef222b3 100644
--- a/cs_gan/main_cs.py
+++ b/cs_gan/main_cs.py
@@ -22,7 +22,7 @@ from absl import app
 from absl import flags
 from absl import logging
 
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 import tensorflow_probability as tfp
 
 from cs_gan import cs
diff --git a/cs_gan/nets.py b/cs_gan/nets.py
index b6ccb41..7291dea 100644
--- a/cs_gan/nets.py
+++ b/cs_gan/nets.py
@@ -17,7 +17,7 @@ import functools
 import re
 import numpy as np
 import sonnet as snt
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 import tensorflow_gan as tfgan
 
 
diff --git a/cs_gan/tests/gan_test.py b/cs_gan/tests/gan_test.py
index 11d30df..8f91eed 100644
--- a/cs_gan/tests/gan_test.py
+++ b/cs_gan/tests/gan_test.py
@@ -14,7 +14,7 @@
 
 
 import sonnet as snt
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 
 from cs_gan import gan
 
diff --git a/cs_gan/utils.py b/cs_gan/utils.py
index adb5ea4..7685bd2 100644
--- a/cs_gan/utils.py
+++ b/cs_gan/utils.py
@@ -19,7 +19,7 @@ import os
 
 from absl import logging
 import numpy as np
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 import tensorflow_probability as tfp
 
 from cs_gan import nets
diff --git a/curl/layers.py b/curl/layers.py
index 2b13e57..de8c8a6 100644
--- a/curl/layers.py
+++ b/curl/layers.py
@@ -17,7 +17,7 @@
 
 from absl import logging
 import sonnet as snt
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 
 tfc = tf.compat.v1
 
diff --git a/curl/model.py b/curl/model.py
index 39dc28f..987d6fa 100644
--- a/curl/model.py
+++ b/curl/model.py
@@ -18,7 +18,7 @@
 from absl import logging
 import numpy as np
 import sonnet as snt
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 import tensorflow_probability as tfp
 
 from curl import layers
diff --git a/curl/training.py b/curl/training.py
index 086d068..37cddbd 100644
--- a/curl/training.py
+++ b/curl/training.py
@@ -22,7 +22,7 @@ from absl import logging
 import numpy as np
 from sklearn import neighbors
 import sonnet as snt
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 import tensorflow_datasets as tfds
 import tensorflow_probability as tfp
 
diff --git a/curl/utils.py b/curl/utils.py
index 33ea298..7611531 100644
--- a/curl/utils.py
+++ b/curl/utils.py
@@ -20,7 +20,7 @@ from __future__ import division
 from __future__ import print_function
 
 from absl import logging
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 import tensorflow_probability as tfp
 
 
diff --git a/hierarchical_probabilistic_unet/geco_utils.py b/hierarchical_probabilistic_unet/geco_utils.py
index aa770d7..ebe5a5f 100644
--- a/hierarchical_probabilistic_unet/geco_utils.py
+++ b/hierarchical_probabilistic_unet/geco_utils.py
@@ -23,7 +23,7 @@ from __future__ import print_function
 
 import numpy as np
 import sonnet as snt
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 
 
 class MovingAverage(snt.AbstractModule):
diff --git a/hierarchical_probabilistic_unet/model_test.py b/hierarchical_probabilistic_unet/model_test.py
index 3f8e2aa..77ff8db 100644
--- a/hierarchical_probabilistic_unet/model_test.py
+++ b/hierarchical_probabilistic_unet/model_test.py
@@ -19,7 +19,7 @@ from __future__ import division
 from __future__ import print_function
 
 from model import HierarchicalProbUNet
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 
 
 _NUM_CLASSES = 2
diff --git a/hierarchical_probabilistic_unet/unet_utils.py b/hierarchical_probabilistic_unet/unet_utils.py
index 61c67e7..25355b0 100644
--- a/hierarchical_probabilistic_unet/unet_utils.py
+++ b/hierarchical_probabilistic_unet/unet_utils.py
@@ -19,7 +19,7 @@ from __future__ import division
 from __future__ import print_function
 
 import sonnet as snt
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 
 
 def res_block(input_features, n_channels, n_down_channels=None,
diff --git a/transporter/transporter.py b/transporter/transporter.py
index 406e48f..cb14e90 100644
--- a/transporter/transporter.py
+++ b/transporter/transporter.py
@@ -21,7 +21,7 @@ from __future__ import print_function
 import functools
 
 import sonnet as snt
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 from tensorflow.contrib import framework as contrib_framework
 from tensorflow.contrib import layers as contrib_layers
 
diff --git a/transporter/transporter_test.py b/transporter/transporter_test.py
index c18ef6c..56f1e07 100644
--- a/transporter/transporter_test.py
+++ b/transporter/transporter_test.py
@@ -19,7 +19,7 @@ from __future__ import division
 from __future__ import print_function
 
 from absl.testing import parameterized
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 from transporter import transporter
 
 
diff --git a/tvt/losses.py b/tvt/losses.py
index d83e352..8797e61 100644
--- a/tvt/losses.py
+++ b/tvt/losses.py
@@ -21,7 +21,7 @@ from __future__ import division
 from __future__ import print_function
 
 import six
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 
 
 def sum_time_average_batch(tensor, name=None):
diff --git a/tvt/main.py b/tvt/main.py
index c81b123..0b1f2d5 100644
--- a/tvt/main.py
+++ b/tvt/main.py
@@ -28,7 +28,7 @@ from absl import logging
 import numpy as np
 from six.moves import range
 from six.moves import zip
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 
 from tvt import batch_env
 from tvt import nest_utils
diff --git a/tvt/memory.py b/tvt/memory.py
index 56bc80a..4ab2ca4 100644
--- a/tvt/memory.py
+++ b/tvt/memory.py
@@ -23,7 +23,7 @@ from __future__ import print_function
 import collections
 
 import sonnet as snt
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 
 ReadInformation = collections.namedtuple(
     'ReadInformation', ('weights', 'indices', 'keys', 'strengths'))
diff --git a/tvt/rma.py b/tvt/rma.py
index dccbd32..9027b48 100644
--- a/tvt/rma.py
+++ b/tvt/rma.py
@@ -26,7 +26,7 @@ import numpy as np
 from six.moves import range
 from six.moves import zip
 import sonnet as snt
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 import trfl
 
 from tvt import losses
diff --git a/unrestricted_advx/main.py b/unrestricted_advx/main.py
index 3c39083..00367a9 100644
--- a/unrestricted_advx/main.py
+++ b/unrestricted_advx/main.py
@@ -16,7 +16,7 @@
 from __future__ import absolute_import
 from __future__ import division
 from __future__ import print_function
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 import tensorflow_hub as hub
 from unrestricted_advex import eval_kit
 
diff --git a/unsupervised_adversarial_training/quick_eval_cifar.py b/unsupervised_adversarial_training/quick_eval_cifar.py
index 7fd21a5..249a628 100644
--- a/unsupervised_adversarial_training/quick_eval_cifar.py
+++ b/unsupervised_adversarial_training/quick_eval_cifar.py
@@ -29,7 +29,7 @@ from absl import flags
 import cleverhans
 from cleverhans import attacks
 import numpy as np
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 from tensorflow.python.ops import math_grad
 import tensorflow_hub as hub