seq2seq-attn icon indicating copy to clipboard operation
seq2seq-attn copied to clipboard

Disable CUDA cross-GPU checks when 2 GPUs are used

Open macournoyer opened this issue 8 years ago • 0 comments

Fixes the following error when training on two GPUs:

Assertion `THCTensor_(checkGPU)(state, 3, self_, src1, src2)'

Solution found here: https://github.com/torch/cutorch/issues/434#issuecomment-237281807

macournoyer avatar Dec 28 '16 14:12 macournoyer