Torchrec for older pytorch and fbgemm

[ali-fani-sd]

Hi,

I’m trying to run a batch transform job on SageMaker, but I’m encountering an issue: the SageMaker instance uses an older NVIDIA driver that only supports CUDA runtime version 11.4. This limits me to using PyTorch versions 2.0.x or 2.2.0. Given that, which version of TorchRec should I use to ensure compatibility with both PyTorch and FBGEMM? I’ve confirmed that FBGEMM version 0.5.0 is compatible for torch 2.0.x. Can you give me for both 2.0.x or 2.2.x

Thanks