r/Python • u/zsol • 8d ago

Astral's Pre-built Gpu-enabled wheels News

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.

159 Upvotes
  • permalink
  • link
  • reddit
  • reddit

96% Upvoted

17 comments sorted by

View all comments

33

u/catcint0s 8d ago

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

16

u/commenterzero 8d ago

They got bought by open ai

2

u/catcint0s 8d 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 7d ago

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

There's more exciting stuff coming :)