ASL icon indicating copy to clipboard operation
ASL copied to clipboard

Published 20 hours ago • verified publisher icon Alibaba-MIIL

How to multi-GPU training

Open ckx506772099 opened this issue 2 years ago • 0 comments

The following error occurred in anti_aliasing.py:40 when I used multiple Gpus for training

RuntimeError: Assertion `THCTensor_(checkGPU)(state, 3, input, output, weight)' failed. Some of weight/gradient/input tensors are located on different GPUs. Please move them to a single one. at /pytorch/aten/src/THCUNN/generic/SpatialDepthwiseConvolution.cu:16

ckx506772099 avatar Jun 22 '22 13:06 ckx506772099