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

Update pytorch version #233

Merged
merged 1 commit into from
Mar 23, 2021
Merged

Conversation

ajschmidt8
Copy link
Member

This PR pins pytorch to version 1.7. The reason this is being updated to 1.7 is because 1.7 supports CUDA 11.0 (src) and 1.8 does not (src).

@ajschmidt8 ajschmidt8 requested a review from a team as a code owner March 23, 2021 16:46
@ajschmidt8
Copy link
Member Author

failures are related to CUDA 11.2, which isn't relevant for this PR. admin merging.

@ajschmidt8 ajschmidt8 merged commit 1ea6965 into rapidsai:branch-0.19 Mar 23, 2021
@ajschmidt8 ajschmidt8 deleted the pytorch-version branch March 23, 2021 17:45
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants