Exception: module 'triton' has no attribute 'set_allocator'

[TianTengya]

my env as follow:

torch==2.6.0
fbgemm_gpu==1.1.0
torchrec==1.1.0
gin_config==0.5.0
pandas==2.2.0
tensorboard==2.19.0
pybind11

[TianTengya]

There seems to be an error with the Triton version. Try temporarily checking out commit f7316fc.

[lijinlei]

I had the same problem? So did you solve the issue?