Added docs

experimenti · Jul 12, 2018 · 90403f9 · 90403f9
1 parent ce7e5e1
commit 90403f9
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -191,6 +191,12 @@ multiproc will open a log for all processes other than the main process.
 
 We suggest using the gloo backend which defaults to TCP if Infiniband isn't available. Using NCCL2 is also possible as a backend. More information [here](http:https://pytorch.org/docs/master/distributed.html#distributed-basics).
 
+You can also specify specific GPU IDs rather than allowing the script to use all available GPUs:
+
+```
+python -m multiproc train.py --visdom --cuda --device-ids 0,1,2,3 # Add your parameters as normal, will only run on 4 GPUs
+```
+
 ### Noise Augmentation/Injection
 
 There is support for two different types of noise; noise augmentation and noise injection.