Skip to content

Commit

Permalink
Merge pull request #85 from chainer/fix-doc-for-mnist-gpu-option
Browse files Browse the repository at this point in the history
Mention --gpu option in the MNIST tutorial
  • Loading branch information
shu65 committed Jun 8, 2017
2 parents 93bd059 + b982b2b commit ead75dc
Showing 1 changed file with 12 additions and 1 deletion.
13 changes: 12 additions & 1 deletion docs/source/tutorial/step1_communicators_optimizers.rst
Original file line number Diff line number Diff line change
Expand Up @@ -63,11 +63,22 @@ Run

With the above two changes, your script is ready for distributed
training. Invoke your script with ``mpiexec`` or ``mpirun`` (see your
MPI's manual for details). The following is an example to execute the
MPI's manual for details). The following is an example of executing the
training with four processes at localhost::

$ mpiexec -n 4 python train_mnist.py

In the non-GPU mode, you may see a warning like shown below,
but this message is harmless, and you can ignore it for now ::

Warning: using naive communicator because only naive supports CPU-only execution


If you have multiple GPUs on the localhost, 4 for example, you
may also want to try::

$ mpiexec -n 4 python train_mnist.py --gpu


Multi-node execution
~~~~~~~~~~~~~~~~~~~~
Expand Down

0 comments on commit ead75dc

Please sign in to comment.