浏览代码

_

/exp-vince/sep30-2020
vincentpierre 4 年前
当前提交
3e978ab7
共有 1 个文件被更改,包括 3 次插入1 次删除
  1. 4
      ml-agents/mlagents/torch_utils/cpu_utils.py

4
ml-agents/mlagents/torch_utils/cpu_utils.py


By default, PyTorch uses 1/2 of the available cores.
"""
num_cpus = _get_num_available_cpus()
print("\n\n\n\n\n NUMCPU ", num_cpus, "\n\n\n\n\n")
print("NUM_CPU ", num_cpus)
logger.info("\n\n\n\n\n NUMCPU " + str(num_cpus) + "\n\n\n\n\n")
return max(min(num_cpus // 2, 4), 1) if num_cpus is not None else None

for Docker containers that are limited in cores.
"""
period = _read_in_integer_file("/sys/fs/cgroup/cpu/cpu.cfs_period_us")
print("period ", period)
print("quota ", quota)
if period > 0 and quota > 0:
return int(quota // period)
else:

正在加载...
取消
保存