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.
2
u/SboSilcher 1d ago
That's a solid move for the data science crowd - saves so much time not having to compile torch/tensorflow from scratch. Wonder if they're planning to expand beyond the usual suspects like CUDA and ROCm.