Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Data parallelism across multiple GPUs #121

Open
wants to merge 7 commits into
base: master
from
Open

Conversation

@dennybritz
Copy link
Contributor

dennybritz commented Mar 28, 2017

Allow the user to replicate the model on multiple GPUs. Still WIP and untested.

@dennybritz
Copy link
Contributor Author

dennybritz commented Mar 28, 2017

This code works, but it is currently very slow. Need to very the op placement on different GPUs to figure out why it is slow.

@dennybritz
Copy link
Contributor Author

dennybritz commented Mar 28, 2017

Ref #44

@dennybritz
Copy link
Contributor Author

dennybritz commented Mar 29, 2017

A few things:

  • Variables need to be placed on CPU
  • Optimizer ops and preprocessing need to be on CPU

It's probably cleaner to put this into the Estimator class. For example, subclass estimator and add support for model replicas.

@bhack
Copy link

bhack commented Apr 11, 2017

@SvensBigData
Copy link

SvensBigData commented May 4, 2017

I get the following error on this branch:

InvalidArgumentError (see above for traceback): Cannot assign a device to node 'save/ShardedFilename_1': Could not satisfy explicit device specification '/device:GPU:1' because no supported kernel for GPU devices is available.
Colocation Debug Info:
Colocation group had the following types and devices:
Identity: CPU
ShardedFilename: CPU
[[Node: save/ShardedFilename_1 = ShardedFilename[_device="/device:GPU:1"](save/StringJoin, save/ShardedFilename_1/shard, save/num_shards)]]

@wabbitml
Copy link

wabbitml commented May 10, 2017

@dennybritz : wanted to know if the branch is usable now. If there are some specific ToDos I can help with the implementation

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Linked issues

Successfully merging this pull request may close these issues.

None yet

4 participants
You can’t perform that action at this time.