diff --git a/CSFM.py b/CSFM.py
index 60fb73c..152301f 100644
--- a/CSFM.py
+++ b/CSFM.py
@@ -1,4 +1,4 @@
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 from utils import gradient_sensitive_loss, tf_ms_ssim
 
 class Model(object):
diff --git a/ESPCN.py b/ESPCN.py
index 8b290a0..060f4e1 100644
--- a/ESPCN.py
+++ b/ESPCN.py
@@ -1,4 +1,4 @@
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 from utils import gradient_sensitive_loss, tf_ms_ssim
 
 class Model(object):
diff --git a/FSRCNN.py b/FSRCNN.py
index 31884e5..8ae5c45 100644
--- a/FSRCNN.py
+++ b/FSRCNN.py
@@ -1,4 +1,4 @@
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 from utils import gradient_sensitive_loss, tf_ms_ssim
 
 class Model(object):
diff --git a/LapSRN.py b/LapSRN.py
index 11be706..78d7e1b 100644
--- a/LapSRN.py
+++ b/LapSRN.py
@@ -1,4 +1,4 @@
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 import math
 from utils import bilinear_upsample_weights, bicubic_downsample
 
diff --git a/main.py b/main.py
index be7ba29..7cb847e 100644
--- a/main.py
+++ b/main.py
@@ -1,7 +1,7 @@
 from model import Model
 
 import numpy as np
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 
 import pprint
 import os
diff --git a/model.py b/model.py
index c4c3240..5c56904 100644
--- a/model.py
+++ b/model.py
@@ -12,7 +12,7 @@ import importlib
 from random import randrange
 
 import numpy as np
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 
 from PIL import Image
 import pdb
diff --git a/utils.py b/utils.py
index 7c7d2c1..f30ad24 100644
--- a/utils.py
+++ b/utils.py
@@ -9,7 +9,7 @@ import subprocess
 import io
 from random import randrange, shuffle
 
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 from PIL import Image
 import numpy as np
 import multiprocessing