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 |
|