GpuManager assumes there are 8 GPUs?

### 🐛 Describe the bug

https://github.com/meta-pytorch/torchforge/blob/main/src/forge/controller/provisioner.py#L122

Should this use torch.cuda.device_count() or something of that sort? 

### Versions

_No response_