[FEA] stable version dependency

[jq]

Is your feature request related to a problem? Please describe.
The current torch version in our image is 2.6. We need to verify whether PyTorch Lightning is compatible with this version. PyTorch Lightning is currently at version 2.5.1, so compatibility may not be an issue since Lightning typically keeps up with the latest Torch releases. However, if we plan to use it in production, it’s better to ensure version stability.

Describe the solution you'd like

best solution is enable pip install, but it has to wait till torchrec release.

Describe alternatives you've considered
temp solution is let's fix it with 2.6

Additional context
Add any other context, code examples, or references to existing implementations about the feature request here.

---

By submitting this issue, you agree to follow our code of conduct and our contributing guidelines.

[shijieliu]

Thanks. Will switch to pip install after torchrec and fbgemm releases.

[JacoCheung]

Update: TorchRec 1.2.0 are released.

[shijieliu]

thanks @JacoCheung , I will update the dockerfile

[shijieliu]

@shijieliu currently encountered undefined symbol error when installing fbgemm_gpu using pip.

[shijieliu]

We have upgraded the dependency to torchrec 1.2.0 and fbgemm_gpu 1.2.0 in #74 .