m53297601
/
Paddle
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Merge pull request #5828 from tensor-tang/develop

Browse Source 
fix v2 init issue on Mac
release/0.11.0
tensor-tang 7 years ago committed by GitHub
parent 657776012b 32eb0a7fcf
commit e4397c4aa8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File

4
python/paddle/v2/__init__.py
Unescape View File

@ -91,14 +91,14 @@ def set_omp_mkl_env_vars(trainer_count):
.read()) .read())
return num_sockets * num_cores_per_socket return num_sockets * num_cores_per_socket
else: else:
cmds = {"Darwin": "sysctl hw.physicalcpu"} cmds = {"Darwin": "sysctl -n hw.physicalcpu"}
return int(os.popen(cmds.get(platform.system(), "expr 1")).read()) return int(os.popen(cmds.get(platform.system(), "expr 1")).read())
def num_logical_processors(): def num_logical_processors():
'''Get the number of logical processors''' '''Get the number of logical processors'''
cmds = { cmds = {
"Linux": "grep \"processor\" /proc/cpuinfo|sort -u|wc -l", "Linux": "grep \"processor\" /proc/cpuinfo|sort -u|wc -l",
"Darwin": "sysctl hw.logicalcpu" "Darwin": "sysctl -n hw.logicalcpu"
} }
return int(os.popen(cmds.get(platform.system(), "expr 1")).read()) return int(os.popen(cmds.get(platform.system(), "expr 1")).read())

Write Preview
Loading…
Cancel
Save