Prebuilt wheels for PyTorch packages with custom ops
I've created a repository that can build PyTorch wheels with custom ops through the GitHub Actions pipeline and publish them using GitHub Releases, with build attestation. Check it out at https://github.com/MiroPsota/torch_packages_builder.
Since there are various ways how to use it, please refer to the repository README for more information.
If you prefer own build or can't trust a 3rd party repository, feel free to fork it and build any package/version/commit ID you desire yourself.
- No Support for Pip Cache: pip relies on http cache, and GitHub generates on-the-fly redirections for release links, so they are probably not playing nicely together. I recommend hosting it yourself.
fused_ssim specific info:
Install using pip:
pip install --extra-index-url https://miropsota.github.io/torch_packages_builder fused_ssim==<version>+<OPTIONAL_commit_hash>pt<PyTorch_version><compute_platform>
Where <compute_platform> is, as in PyTorch, one of cpu, cu<CUDA_short_version> (e.g. cu128, cu121, cu118), or rocm<ROCM_version> (not supported right now).
For example, the newest fused_ssim commit (as of writing) 30fb258, PyTorch 2.8.0 with CUDA 12.9:
pip install --extra-index-url https://miropsota.github.io/torch_packages_builder fused_ssim==0.0.1+30fb258pt2.8.0cu129
Look at releases section if there are any other combinations, I will probably build occasionally with new pytorch releases and versions/commits.
These wheels are built with PyTorch versions 2.3.0 to 2.8.0 and their respective compute platforms and supported operating systems. Please note an exception for the ROCm platform. The build is done using the available public GitHub runners, so they might not work on older OSes.
Although the wheels have been successfully built, I don't test them to work correctly.
If you've installed PyTorch with pip, there's no need to have CUDA installed on your system, as the PyTorch wheels for pip bundle CUDA.
Prebuilt wheels for PyTorch packages with custom ops
I've created a repository that can build PyTorch wheels with custom ops through the GitHub Actions pipeline and publish them using GitHub Releases, with build attestation. Check it out at https://github.com/MiroPsota/torch_packages_builder.
Since there are various ways how to use it, please refer to the repository README for more information.
If you prefer own build or can't trust a 3rd party repository, feel free to fork it and build any package/version/commit ID you desire yourself.
fused_ssim specific info:
Install using pip:
Where
<compute_platform>is, as in PyTorch, one ofcpu,cu<CUDA_short_version>(e.g.cu128,cu121,cu118), orrocm<ROCM_version>(not supported right now).For example, the newest fused_ssim commit (as of writing)
30fb258, PyTorch 2.8.0 with CUDA 12.9:Look at releases section if there are any other combinations, I will probably build occasionally with new pytorch releases and versions/commits.
These wheels are built with PyTorch versions
2.3.0to2.8.0and their respective compute platforms and supported operating systems. Please note an exception for the ROCm platform. The build is done using the available public GitHub runners, so they might not work on older OSes.Although the wheels have been successfully built, I don't test them to work correctly.
If you've installed PyTorch with pip, there's no need to have CUDA installed on your system, as the PyTorch wheels for pip bundle CUDA.