diff --git a/tensorflow/tools/test/system_info_lib.py b/tensorflow/tools/test/system_info_lib.py index 0cc261591bc..59a30f9a368 100644 --- a/tensorflow/tools/test/system_info_lib.py +++ b/tensorflow/tools/test/system_info_lib.py @@ -81,7 +81,7 @@ def gather_cpu_info(): # Gather num_cores_allowed try: with gfile.GFile('/proc/self/status', 'rb') as fh: - nc = re.search(r'(?m)^Cpus_allowed:\s*(.*)$', fh.read()) + nc = re.search(r'(?m)^Cpus_allowed:\s*(.*)$', fh.read().decode('utf-8')) if nc: # e.g. 'ff' => 8, 'fff' => 12 cpu_info.num_cores_allowed = ( bin(int(nc.group(1).replace(',', ''), 16)).count('1'))