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?

[bepluvia]

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

Me too. Has this problem been solved well?

[bepluvia]

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

Me too. Has this problem been solved well?

I updated the torch version to 2.7.0. Since triton is closely related to torch, it was automatically updated to 3.3.0 as well. This version can run successfully (previously it was torch==2.6.0).

This is my current pip list:

fbgemm_gpu 1.2.0
gin-config 0.5.0
pandas 2.2.0
tensorboard 2.19.0
torch 2.7.0+cu118
torchrec 1.2.0
triton 3.3.0