Skip to content
This repository has been archived by the owner on Jun 10, 2021. It is now read-only.

multi-GPU #538

Open
wanghechong opened this issue Apr 17, 2018 · 2 comments
Open

multi-GPU #538

wanghechong opened this issue Apr 17, 2018 · 2 comments

Comments

@wanghechong
Copy link

wanghechong commented Apr 17, 2018

i want to use multi-GPU to train my model, but it report error like"torch/install/share/lua/5.1/cunn/DataParallelTable.lua:348: attempt to call method 'type' (a nil value)"
i do not know why, but the cunn is our package dowloaded from net

@guillaumekln
Copy link
Collaborator

What code are you running? OpenNMT does not use nn.DataParallelTable.

@wanghechong
Copy link
Author

the problem is that i do language generation, and use a mlp encoder and a LSTM decoder, because i expecially want to test the big batch so i have to train model on two GPUs because of out of memory.
so i have tried it many times, i found DataParallelTable is not suitable for my situation, i see someone use cutorch.setDevice() but i do not know very clearly about his logic relation, so i hope you can give me some advice!!
@

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Development

No branches or pull requests

2 participants