Merge pull request #9 from adujardin/fix_cudnn_error

Fix possible CUDNN_STATUS_INTERNAL_ERROR on RTX cards

Author: CalciferZh

wrappers.py

@@ -23,7 +23,9 @@ def __init__(self, model_path):
     self.graph = tf.Graph()
     with self.graph.as_default():
       with tf.variable_scope('prior_based_hand'):
-        self.sess = tf.Session()
+        config = tf.ConfigProto()
+        config.gpu_options.allow_growth = True
+        self.sess = tf.Session(config=config)
         self.input_ph = tf.placeholder(tf.uint8, [128, 128, 3])
         self.feed_img = \
           tf.cast(tf.expand_dims(self.input_ph, 0), tf.float32) / 255
@@ -93,7 +95,9 @@ def __init__(self, input_size, network_fn, model_path, net_depth, net_width):
       with tf.name_scope('network'):
         self.theta = \
           network_fn(self.input_ph, net_depth, net_width, training=False)[0]
-      self.sess = tf.Session()
+        config = tf.ConfigProto()
+        config.gpu_options.allow_growth = True
+        self.sess = tf.Session(config=config)
       tf.train.Saver().restore(self.sess, model_path)
 
   def process(self, joints):