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

environment.yml: install pytorch with gpu support #545

Merged
merged 2 commits into from
Jun 6, 2022

Conversation

remtav
Copy link
Contributor

@remtav remtav commented May 26, 2022

fixes #529
see #540

@remtav remtav changed the title environment.yml: peg cudatoolkit to max v11.5 environment.yml: install pytorch with gpu support by pegging cudatoolkit to max v11.5 May 26, 2022
environment.yml Outdated Show resolved Hide resolved
Co-authored-by: Wei Ji <[email protected]>
Copy link
Collaborator

@adamjstewart adamjstewart left a 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?

@calebrob6
Copy link
Member

Nice find @weiji14! I just confirmed this works (with cuda support) on Ubuntu

@adamjstewart adamjstewart marked this pull request as ready for review June 6, 2022 17:34
@adamjstewart adamjstewart merged commit 60b7a3d into microsoft:main Jun 6, 2022
@weiji14
Copy link
Contributor

weiji14 commented Jun 6, 2022

Should have changed the PR title before merging 😅

@adamjstewart adamjstewart changed the title environment.yml: install pytorch with gpu support by pegging cudatoolkit to max v11.5 environment.yml: install pytorch with gpu support Jun 6, 2022
@adamjstewart
Copy link
Collaborator

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.

@remtav remtav deleted the 529-conda-pytorch-cpu2 branch July 6, 2022 17:31
@adamjstewart adamjstewart added this to the 0.3.0 milestone Jul 10, 2022
@adamjstewart adamjstewart mentioned this pull request Jul 11, 2022
yichiac pushed a commit to yichiac/torchgeo that referenced this pull request Apr 29, 2023
…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]>
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.

Conda installs a cpu version of pytorch and torchvision
4 participants