-
Notifications
You must be signed in to change notification settings - Fork 370
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
environment.yml: install pytorch with gpu support #545
Conversation
Co-authored-by: Wei Ji <[email protected]>
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM. Can someone confirm that this now works as intended?
Nice find @weiji14! I just confirmed this works (with cuda support) on Ubuntu |
Should have changed the PR title before merging 😅 |
Oh well, commit messages aren't super important to me, issue titles are more important because that's what I look for when writing release notes. I think this change is minor enough that it won't be in the release notes anyway. |
…kit to max v11.5 (microsoft#545) * environment.yml: peg cudatoolkit to max v11.5 * remove cudatoolkit Co-authored-by: Wei Ji <[email protected]> Co-authored-by: Wei Ji <[email protected]>
fixes #529
see #540