No module named 'triton.third_party'

[Yezhibin701227]

i install the packages of torch==2.1.2+cu118 and xformers==0.0.23.post1 in the conda, follow the steps of "Alpaca+Llama-3 8B full example.ipynb" to run on the local machine with an A6000 GPU. The error info attached with file.

[Yezhibin701227]

I found out the problem , reinstalled the xformers:
pip3 install -U xformers==0.0.23.post1 --index-url https://download.pytorch.org/whl/cu118

[danielhanchen]

Oh great you solved it!