paddle使用cpu正常,改用gpu时报错 RuntimeError: parallel_for failed: invalid configuration argument
Created by: shaofeifei11
paddle-gpu: 1.8.4.post107 paddle: 1.8.1
部分代码: use_cuda = True place = fluid.CUDAPlace(0) if use_cuda else fluid.CPUPlace() exe = fluid.Executor(place) start_program = fluid.default_startup_program() exe.run(start_program) # 一定要在 exe.run(start_program) 之后才能执行load_persistables加载保存的数据