未验证 提交 383bde0b 编写于 作者: W wangguanzhong 提交者: GitHub

fix multi-gpu training in static (#4835)

上级 511d179b
...@@ -112,7 +112,7 @@ def main(): ...@@ -112,7 +112,7 @@ def main():
if cfg.use_gpu: if cfg.use_gpu:
devices_num = fluid.core.get_cuda_device_count() devices_num = fluid.core.get_cuda_device_count()
if cfg.use_npu: elif cfg.use_npu:
devices_num = fluid.core.get_npu_device_count() devices_num = fluid.core.get_npu_device_count()
elif use_xpu: elif use_xpu:
# ToDo(qingshu): XPU only support single card now # ToDo(qingshu): XPU only support single card now
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册