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.

Sign up for GitHub

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

Jump to bottom

Remove cuda gencode 90 to reduce onnxruntime-training package size #19486

Merged
merged 1 commit into from
Feb 12, 2024
Merged

Remove cuda gencode 90 to reduce onnxruntime-training package size #19486

merged 1 commit into from
Feb 12, 2024

Conversation

baijumeswani
Copy link
Contributor

@baijumeswani baijumeswani commented Feb 10, 2024 •
edited
Loading

onnxruntime-training CUDA 11.8 package size is > 300 MB and as a result publishing to PyPI fails.
gencode 90 requires CUDA > 12.

This change brings the package size < 300 MB.

@baijumeswani baijumeswani added training issues related to ONNX Runtime training; typically submitted using template build build issues; typically submitted using template labels Feb 10, 2024
@baijumeswani baijumeswani requested a review from a team as a code owner February 10, 2024 02:04
@baijumeswani baijumeswani merged commit c831031 into main Feb 12, 2024
97 of 99 checks passed
@baijumeswani baijumeswani deleted the baijumeswani/remove-cuda-gencode branch February 12, 2024 17:24
YUNQIUGUO pushed a commit that referenced this pull request Feb 12, 2024
@baijumeswani
Remove cuda gencode 90 to reduce onnxruntime-training package size (#…
14543de 
…19486)
@sophies927 sophies927 added the triage:approved Approved for cherrypicks for release label Feb 13, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@snnn snnn snnn approved these changes

@askhade askhade Awaiting requested review from askhade

@pengwa pengwa Awaiting requested review from pengwa

Assignees
No one assigned
Labels
build build issues; typically submitted using template release:1.17.1 training issues related to ONNX Runtime training; typically submitted using template triage:approved Approved for cherrypicks for release
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
3 participants
@baijumeswani @snnn @sophies927