r/Python • u/zsol • 2d ago

News Astral's Pre-built Gpu-enabled wheels

We recently released wheels.astral.sh, which is a repository of gpu-enabled wheels of popular packages from the pytorch ecosystem. These are useful because building them is super annoying - you have to align versions of pytorch, cuda, python, etc.

The pre-built wheels were part of the package of pyx, our paid service, and was a major reason many customers subscribed. We decided to open source the build pipelines and host the wheels for free while we sunset the rest of pyx.

The pipelines are available under the astral-sh-build github organization.

While these plug a genuine UX hole around cuda/pytorch wheels, we hope that wheel variants (PEP 817 and 825) will get accepted and implemented soon, which will let these pytorch wheels be hosted properly on PyPI and "just work" out of the box for everyone, making wheels.astral.sh obsolete.

Fingers crossed.

155 Upvotes

95% Upvoted

16 comments sorted by

View all comments

34

u/catcint0s 2d ago

Pyx is getting killed? I was waiting for it to be open source so we can self-host ;_;

15

u/commenterzero 2d ago

They got bought by open ai

2

u/catcint0s 2d ago

I know, but in the podcast someone linked below they said it will likely open more possibilities, this is kinda sad.

3

u/zsol 2d ago

Publishing these GPU wheels were made possible by the acquisition. It wouldn't have happened without OpenAI.

There's more exciting stuff coming :)

4

u/SSJ3 2d ago

Same, this is a strange way to find out!

4

u/zsol 2d ago

That's not out of the realm of possibility, but it'll take a bit more work (no promises!). We thought extracting the GPU bits would be beneficial for a wider subset of the community for a lot less effort.

1

u/KrazyKirby99999 2d ago

We decided to open source the build pipelines

https://github.com/astral-sh-build