提交 191d8dce 编写于 作者: Q qijun

set nccl = true

上级 7dbdf007
...@@ -31,7 +31,7 @@ use_nccl = False ...@@ -31,7 +31,7 @@ use_nccl = False
place = fluid.CPUPlace() place = fluid.CPUPlace()
if fluid.core.is_compiled_with_cuda(): if fluid.core.is_compiled_with_cuda():
device_type = 'CUDA' device_type = 'CUDA'
use_nccl = False use_nccl = True
place = fluid.CUDAPlace(0) place = fluid.CUDAPlace(0)
places = fluid.layers.get_places(device_count=0, device_type=device_type) places = fluid.layers.get_places(device_count=0, device_type=device_type)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册