please use tools/python/tf_sess_config.py to get the sess_config after that. note the byte order of session config is in normal order. bump the MICRO version for the config change.
Showing the changes to tools/python (1 of 3 files). Show all
Unified · Split
| @@ -0,0 +1,45 @@ | |||
| 1 | # Copyright (c) 2021 | ||
| 2 | # | ||
| 3 | # This file is part of FFmpeg. | ||
| 4 | # | ||
| 5 | # FFmpeg is free software; you can redistribute it and/or | ||
| 6 | # modify it under the terms of the GNU Lesser General Public | ||
| 7 | # License as published by the Free Software Foundation; either | ||
| 8 | # version 2.1 of the License, or (at your option) any later version. | ||
| 9 | # | ||
| 10 | # FFmpeg is distributed in the hope that it will be useful, | ||
| 11 | # but WITHOUT ANY WARRANTY; without even the implied warranty of | ||
| 12 | # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | ||
| 13 | # Lesser General Public License for more details. | ||
| 14 | # | ||
| 15 | # You should have received a copy of the GNU Lesser General Public | ||
| 16 | # License along with FFmpeg; if not, write to the Free Software | ||
| 17 | # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA | ||
| 18 | # ============================================================================== | ||
| 19 | |||
| 20 | # verified with Python 3.6.8 on CentOS 7.2 | ||
| 21 | import tensorflow as tf | ||
| 22 | |||
| 23 | visible_device_list = '0' # use , separator for more GPUs like '0, 1' | ||
| 24 | per_process_gpu_memory_fraction = 0.9 # avoid out of memory | ||
| 25 | intra_op_parallelism_threads = 2 # default in tensorflow | ||
| 26 | inter_op_parallelism_threads = 5 # default in tensorflow | ||
| 27 | |||
| 28 | gpu_options = tf.compat.v1.GPUOptions( | ||
| 29 | per_process_gpu_memory_fraction = per_process_gpu_memory_fraction, | ||
| 30 | visible_device_list = visible_device_list, | ||
| 31 | allow_growth = True) | ||
| 32 | |||
| 33 | config = tf.compat.v1.ConfigProto( | ||
| 34 | allow_soft_placement = True, | ||
| 35 | log_device_placement = False, | ||
| 36 | intra_op_parallelism_threads = intra_op_parallelism_threads, | ||
| 37 | inter_op_parallelism_threads = inter_op_parallelism_threads, | ||
| 38 | gpu_options = gpu_options) | ||
| 39 | |||
| 40 | s = config.SerializeToString() | ||
| 41 | # print(list(map(hex, s))) # print by json if need | ||
| 42 | |||
| 43 | print('a serialized protobuf string for TF_SetConfig, note the byte order is in normal order.') | ||
| 44 | b = ''.join(format(b,'02x') for b in s) | ||
| 45 | print('0x%s' % b) # print by hex format | ||