-
Notifications
You must be signed in to change notification settings - Fork 1.3k
Data parallelism across multiple GPUs #121
base: master
Are you sure you want to change the base?
Conversation
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. |
Ref #44 |
A few things:
It's probably cleaner to put this into the |
See also tensorflow/tensorflow#2126 |
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. |
@dennybritz : wanted to know if the branch is usable now. If there are some specific ToDos I can help with the implementation |
Allow the user to replicate the model on multiple GPUs. Still WIP and untested.