Use CPU-only PyTorch wheels on GitHub Actions #5410
Labels
code-fix
Change that does not change the behavior, such as code refactoring.
contribution-welcome
Issue that welcomes contribution.
Motivation
On GitHub Actions environment, PyTorch wheels that contains CUDA-related files are not required. By explicitly opting for the installation of CPU-only wheels, which have smaller file sizes, we can speed up the overall installation process.
Suggestion
Consider adding the
--index-url https://download.pytorch.org/whl/cpu
option to allpip install
commands in GitHub Actions workflow files. It should be noted that some workflows are already utilizing this option, as shown below.See https://pytorch.org/get-started/locally/ for details.
Additional context (optional)
No response
The text was updated successfully, but these errors were encountered: