Not compatible with CUDA 10.2

[liubl1217]

File "/fast/liubl/workspace/HDNet_TikTok/hourglass_net_depth.py", line 119, in hourglass_refinement
out0_d = hourglass_stack_no_incep(netIN)
File "/fast/liubl/workspace/HDNet_TikTok/hourglass_net_depth.py", line 74, in hourglass_stack_no_incep
c0 = conv_layer(stack_in,layer_name('conv'),KER_SZ,NUM_CH[0],bn=bn,training=training)
File "/fast/liubl/workspace/HDNet_TikTok/hourglass_net_depth.py", line 34, in conv_layer
kernel = variable('weights', [kernel_size, kernel_size, input_channels, output_channels], initializer, regularizer=tf.contrib.layers.l2_regularizer(0.0005))
File "/fast/liubl/workspace/HDNet_TikTok/hourglass_net_depth.py", line 28, in variable
return tf.get_variable(name, shape, initializer=initializer, regularizer=regularizer, dtype=tf.float32, trainable=True)
File "/fast/liubl/anaconda3/envs/tiktok2/lib/python3.7/site-packages/tensorflow/python/ops/variable_scope.py", line 1496, in get_variable
aggregation=aggregation)
File "/fast/liubl/anaconda3/envs/tiktok2/lib/python3.7/site-packages/tensorflow/python/ops/variable_scope.py", line 1239, in get_variable
aggregation=aggregation)
File "/fast/liubl/anaconda3/envs/tiktok2/lib/python3.7/site-packages/tensorflow/python/ops/variable_scope.py", line 562, in get_variable
aggregation=aggregation)
File "/fast/liubl/anaconda3/envs/tiktok2/lib/python3.7/site-packages/tensorflow/python/ops/variable_scope.py", line 514, in _true_getter
aggregation=aggregation)
File "/fast/liubl/anaconda3/envs/tiktok2/lib/python3.7/site-packages/tensorflow/python/ops/variable_scope.py", line 929, in _get_single_variable
aggregation=aggregation)
File "/fast/liubl/anaconda3/envs/tiktok2/lib/python3.7/site-packages/tensorflow/python/ops/variables.py", line 259, in call
return cls._variable_v1_call(*args, **kwargs)
File "/fast/liubl/anaconda3/envs/tiktok2/lib/python3.7/site-packages/tensorflow/python/ops/variables.py", line 220, in _variable_v1_call
shape=shape)
File "/fast/liubl/anaconda3/envs/tiktok2/lib/python3.7/site-packages/tensorflow/python/ops/variables.py", line 198, in
previous_getter = lambda **kwargs: default_variable_creator(None, **kwargs)
File "/fast/liubl/anaconda3/envs/tiktok2/lib/python3.7/site-packages/tensorflow/python/ops/variable_scope.py", line 2511, in default_variable_creator
shape=shape)
File "/fast/liubl/anaconda3/envs/tiktok2/lib/python3.7/site-packages/tensorflow/python/ops/variables.py", line 263, in call
return super(VariableMetaclass, cls).call(*args, **kwargs)
File "/fast/liubl/anaconda3/envs/tiktok2/lib/python3.7/site-packages/tensorflow/python/ops/variables.py", line 1568, in init
shape=shape)
File "/fast/liubl/anaconda3/envs/tiktok2/lib/python3.7/site-packages/tensorflow/python/ops/variables.py", line 1698, in _init_from_args
initial_value(), name="initial_value", dtype=dtype)
File "/fast/liubl/anaconda3/envs/tiktok2/lib/python3.7/site-packages/tensorflow/python/ops/variable_scope.py", line 901, in
partition_info=partition_info)
File "/fast/liubl/anaconda3/envs/tiktok2/lib/python3.7/site-packages/tensorflow/contrib/layers/python/layers/initializers.py", line 150, in _initializer
seed=seed)
File "/fast/liubl/anaconda3/envs/tiktok2/lib/python3.7/site-packages/tensorflow/python/ops/random_ops.py", line 178, in truncated_normal
shape_tensor, dtype, seed=seed1, seed2=seed2)
File "/fast/liubl/anaconda3/envs/tiktok2/lib/python3.7/site-packages/tensorflow/python/ops/gen_random_ops.py", line 1013, in truncated_normal
name=name)
File "/fast/liubl/anaconda3/envs/tiktok2/lib/python3.7/site-packages/tensorflow/python/framework/op_def_library.py", line 788, in _apply_op_helper
op_def=op_def)
File "/fast/liubl/anaconda3/envs/tiktok2/lib/python3.7/site-packages/tensorflow/python/util/deprecation.py", line 507, in new_func
return func(*args, **kwargs)
File "/fast/liubl/anaconda3/envs/tiktok2/lib/python3.7/site-packages/tensorflow/python/framework/ops.py", line 3616, in create_op
op_def=op_def)
File "/fast/liubl/anaconda3/envs/tiktok2/lib/python3.7/site-packages/tensorflow/python/framework/ops.py", line 2005, in init
self._traceback = tf_stack.extract_stack()

Seems not support for current CUDA version

[yasaminjafarian]

This code is tested with tensorflow-gpu 1.14.0, Python 3.7.4, CUDA 10 (version 10.0.130) and cuDNN 7 (version 7.4.2)
You can use docker container if there is any other problem https://github.com/yasaminjafarian/HDNet_TikTok#installation