torchrec
torchrec copied to clipboard
Properly set up GPU/CUDA environment for CI
Summary: ATT
Pytorch distributed when writing some GPU unit tests noticed that they didn't have access to GPUs. After some investigation they found that these additional lines were included in the main PyTorch CI. After adding them, we should be able to see GPUs
Differential Revision: D35736829
This pull request was exported from Phabricator. Differential Revision: D35736829
This pull request was exported from Phabricator. Differential Revision: D35736829
This pull request was exported from Phabricator. Differential Revision: D35736829