Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Use CPU-only PyTorch wheels on GitHub Actions #5410

Open
c-bata opened this issue Apr 22, 2024 · 0 comments · May be fixed by #5465
Open

Use CPU-only PyTorch wheels on GitHub Actions #5410

c-bata opened this issue Apr 22, 2024 · 0 comments · May be fixed by #5465
Labels
code-fix Change that does not change the behavior, such as code refactoring. contribution-welcome Issue that welcomes contribution.

Comments

@c-bata
Copy link
Member

c-bata commented Apr 22, 2024

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 all pip install commands in GitHub Actions workflow files. It should be noted that some workflows are already utilizing this option, as shown below.

$ git grep torch .github/
.github/workflows/coverage.yml:        pip install --progress-bar off .[optional] --extra-index-url https://download.pytorch.org/whl/cpu
.github/workflows/performance-benchmarks-bayesmark.yml:        pip install --progress-bar off .[benchmark] --extra-index-url https://download.pytorch.org/whl/cpu
.github/workflows/performance-benchmarks-kurobako.yml:        pip install --progress-bar off .[benchmark] --extra-index-url https://download.pytorch.org/whl/cpu
.github/workflows/performance-benchmarks-mo-kurobako.yml:        pip install --progress-bar off .[benchmark] --extra-index-url https://download.pytorch.org/whl/cpu
.github/workflows/speed-benchmarks.yml:        pip install --progress-bar off .[benchmark] --extra-index-url https://download.pytorch.org/whl/cpu
.github/workflows/sphinx-build.yml:        pip install --progress-bar off -U .[document] --extra-index-url https://download.pytorch.org/whl/cpu
.github/workflows/sphinx-build.yml:        pip install --progress-bar off -U .[document] --extra-index-url https://download.pytorch.org/whl/cpu
.github/workflows/tests-storage.yml:        pip install --progress-bar off .[optional] --extra-index-url https://download.pytorch.org/whl/cpu
.github/workflows/tests-with-minimum-versions.yml:        pip install --progress-bar off .[optional] --extra-index-url https://download.pytorch.org/whl/cpu

See https://pytorch.org/get-started/locally/ for details.

Additional context (optional)

No response

@c-bata c-bata added code-fix Change that does not change the behavior, such as code refactoring. contribution-welcome Issue that welcomes contribution. labels Apr 22, 2024
@porink0424 porink0424 linked a pull request May 29, 2024 that will close this issue
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
code-fix Change that does not change the behavior, such as code refactoring. contribution-welcome Issue that welcomes contribution.
Projects
None yet
Development

Successfully merging a pull request may close this issue.

1 participant