Tongzhou Wang
|
8e33451e2e
|
Make torch.cuda.* take device objects; Update distributed docs (#10833)
Summary:
Commits:
1. Make `torch.cuda.*` take device objects
2. Update `torch.distributed` docs to emphasize calling `torch.cuda.set_device` before `init_process_group`
Pull Request resolved: https://github.com/pytorch/pytorch/pull/10833
Differential Revision: D9514241
Pulled By: SsnL
fbshipit-source-id: 2497464305fb1e63d6c495291a5744aaa7e2696e
|
2018-08-27 15:24:42 -07:00 |
|
Kaiyu Shi
|
db6e4576da
|
Use customized python interpreter (#7520)
|
2018-05-12 13:06:39 -04:00 |
|
Edgar Andrés Margffoy Tuay
|
4ab6ea5b1f
|
Add unbuffered flag to distributed node launcher (#7226)
|
2018-05-03 11:49:06 +02:00 |
|
Teng Li
|
f5beff334b
|
Added distributed docs on NCCL2 backend/functions and launch module (#6579)
|
2018-04-15 21:53:10 -04:00 |
|
Teng Li
|
37059ba0ec
|
Added torch.distributed.launch module for easier multi-proc/node distributed job launching (#5348)
|
2018-03-13 12:04:38 +01:00 |
|