mirror of https://github.com/FFmpeg/FFmpeg.git
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
45 lines
1.9 KiB
45 lines
1.9 KiB
# Copyright (c) 2021 |
|
# |
|
# This file is part of FFmpeg. |
|
# |
|
# FFmpeg is free software; you can redistribute it and/or |
|
# modify it under the terms of the GNU Lesser General Public |
|
# License as published by the Free Software Foundation; either |
|
# version 2.1 of the License, or (at your option) any later version. |
|
# |
|
# FFmpeg is distributed in the hope that it will be useful, |
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of |
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
|
# Lesser General Public License for more details. |
|
# |
|
# You should have received a copy of the GNU Lesser General Public |
|
# License along with FFmpeg; if not, write to the Free Software |
|
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
|
# ============================================================================== |
|
|
|
# verified with Python 3.6.8 on CentOS 7.2 |
|
import tensorflow as tf |
|
|
|
visible_device_list = '0' # use , separator for more GPUs like '0, 1' |
|
per_process_gpu_memory_fraction = 0.9 # avoid out of memory |
|
intra_op_parallelism_threads = 2 # default in tensorflow |
|
inter_op_parallelism_threads = 5 # default in tensorflow |
|
|
|
gpu_options = tf.compat.v1.GPUOptions( |
|
per_process_gpu_memory_fraction = per_process_gpu_memory_fraction, |
|
visible_device_list = visible_device_list, |
|
allow_growth = True) |
|
|
|
config = tf.compat.v1.ConfigProto( |
|
allow_soft_placement = True, |
|
log_device_placement = False, |
|
intra_op_parallelism_threads = intra_op_parallelism_threads, |
|
inter_op_parallelism_threads = inter_op_parallelism_threads, |
|
gpu_options = gpu_options) |
|
|
|
s = config.SerializeToString() |
|
# print(list(map(hex, s))) # print by json if need |
|
|
|
print('a serialized protobuf string for TF_SetConfig, note the byte order is in normal order.') |
|
b = ''.join(format(b,'02x') for b in s) |
|
print('0x%s' % b) # print by hex format
|
|
|