Fix dependencies for the distributed benchmark (#1327)
Summary: Pull Request resolved: https://github.com/pytorch/benchmark/pull/1327
Reviewed By: weatfb
Differential Revision: D41586596
Pulled By: xuzhao9
fbshipit-source-id: 228c4b8ab5c16566d3ba4c4af11bd554480d6628
