Merge pull request #21 from jradrion/dnn_fix

fixed dnn error in predict
kr-colab · May 27, 2020 · a3dd5df · a3dd5df
2 parents 071d6c0 + bbb94c8
commit a3dd5df
Showing 1 changed file with 8 additions and 0 deletions.
diff --git 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")