Torchrec lib - Impossible to execute it on Datalore

What I can do:

  • install Torch and Torchrec module (I’ve tried different versions)

What I cannot do:

  • call Torchrec module which is based on

Code to call class/functions which generate the issue:
import numpy as np
import torch
from import Interaction, ExplicitData
from torchrec.model.implicit import ImplicitMatrixFactorization
from torch.optim import Adam

Error :
Traceback (most recent call last):
AttributeError: ‘_OpNamespace’ ‘fbgemm’ object has no attribute ‘jagged_2d_to_dense’