45 lines · 1.9 KB · Python
RawDownload
# 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.2import 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 memoryintra_op_parallelism_threads = 2  # default in tensorflowinter_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

Keyboard shortcuts

General

Show keyboard shortcuts
?
Focus the current search
/

Issues and pull requests

Open filters
f

Issues, pull requests and Inbox

Select next item
j↓
Select previous item
k↑
Open selected item
Enter
Dismiss filters or clear selection
Esc

Repository code

Go to file
t