提交 62ccb463 编写于 作者: C Chen Weihang

fix cpu num set error

上级 5c5099a9
......@@ -399,8 +399,8 @@ def main(args):
place = fluid.cuda_places()
DEV_COUNT = fluid.core.get_cuda_device_count()
else:
place = fluid.cpu_places()
os.environ['CPU_NUM'] = str(args.cpu_num)
place = fluid.cpu_places()
DEV_COUNT = args.cpu_num
logger.info("Dev Num is %s" % str(DEV_COUNT))
exe = fluid.Executor(place[0])
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册