diff --git a/ReLERNN/helpers.py b/ReLERNN/helpers.py
index 4e409a3..e30cf0e 100644
--- a/ReLERNN/helpers.py
+++ b/ReLERNN/helpers.py
@@ -265,6 +265,14 @@ def load_and_predictVCF(VCFGenerator,
 
     os.environ["CUDA_VISIBLE_DEVICES"] = str(gpuID)
 
+    ## The following code block appears necessary for running with tf2 and cudnn
+    from tensorflow.compat.v1 import ConfigProto
+    from tensorflow.compat.v1 import Session
+    config = ConfigProto()
+    config.gpu_options.allow_growth = True
+    Session(config=config)
+    ###
+
     # load json and create model
     if(network != None):
         jsonFILE = open(network[0],"r")