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

Add support for cuda 11.8 and python 3.11 for training #15548

Merged
merged 8 commits into from
Apr 20, 2023
Merged

Add support for cuda 11.8 and python 3.11 for training #15548

merged 8 commits into from
Apr 20, 2023

Conversation

baijumeswani
Copy link
Contributor

This pull request:

  • Updates the training python package with support for cuda 11.8
  • Adds the training python package with support for python 3.11.
  • Updates the training CI pipeline to work with cuda 11.8
  • Updates the packaging and CI pipeline dependency on torch 2.0

@baijumeswani baijumeswani added the training issues related to ONNX Runtime training; typically submitted using template label Apr 17, 2023
@baijumeswani baijumeswani requested a review from a team as a code owner April 17, 2023 23:50
snnn
snnn previously approved these changes Apr 18, 2023
View reviewed changes
askhade
askhade previously approved these changes Apr 18, 2023
View reviewed changes
@baijumeswani baijumeswani merged commit 11b0a18 into main Apr 20, 2023
@baijumeswani baijumeswani deleted the baijumeswani/training-package branch April 20, 2023 19:56
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@justinchuby justinchuby justinchuby left review comments

@snnn snnn snnn approved these changes

@askhade askhade askhade left review comments

@pengwa pengwa Awaiting requested review from pengwa

Assignees
No one assigned
Labels
training issues related to ONNX Runtime training; typically submitted using template
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.

None yet
4 participants
@baijumeswani @snnn @askhade @justinchuby