jax-ml / jax-triton

jax-triton contains integrations between JAX and OpenAI Triton
Apache License 2.0
340 stars 36 forks source link

Problems in Installing at HEAD #244

Open LeoXinhaoLee opened 10 months ago

LeoXinhaoLee commented 10 months ago

Hi, thank you so much for releasing this wonderful library.

When I was trying to install jax-triton at HEAD, I got this error message: ERROR: triton_nightly-2.1.0.dev20230714011643-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl is not a supported wheel on this platform.

My machine is x86_64 Linux, with the below cuda env

image

Could you please help me with this? Thank you very much for your time and help!

sharadmv commented 10 months ago

Can you check your Python version?

LeoXinhaoLee commented 10 months ago

Thanks for your kind reply. I realized my Python version was 3.9, which could be the problem. Now I've upgraded to 3.10.13, but when installing at HEAD, I got the below error:

image

Could you please help me with this? Thank you very much!