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

Upgrade to pytorch 2.0.1 conda package #1015

Merged
merged 7 commits into from
Jul 5, 2023

Conversation

efajardo-nv
Copy link
Contributor

@efajardo-nv efajardo-nv commented Jun 29, 2023

Description

  • Update docker containers and conda recipe/environments to install pytorch 2.0.1 conda package
  • Also update torch version for training-tuning scripts/notebooks

Closes #1008

Checklist

[x] I am familiar with the Contributing Guidelines.
[x] New or existing tests cover these changes.
[x] The documentation is up to date with these changes.

@efajardo-nv efajardo-nv added enhancement Additional functionality added to an existing feature non-breaking Non-breaking change improvement Improvement to existing functionality and removed enhancement Additional functionality added to an existing feature labels Jun 29, 2023
@review-notebook-app
Copy link

Check out this pull request on  ReviewNB

See visual diffs & provide feedback on Jupyter Notebooks.


Powered by ReviewNB

@efajardo-nv efajardo-nv marked this pull request as ready for review July 5, 2023 18:52
@efajardo-nv efajardo-nv requested review from a team as code owners July 5, 2023 18:52
@jarmak-nv jarmak-nv assigned jarmak-nv and efajardo-nv and unassigned jarmak-nv Jul 5, 2023
@efajardo-nv
Copy link
Contributor Author

/merge

@rapids-bot rapids-bot bot merged commit f71b570 into nv-morpheus:branch-23.07 Jul 5, 2023
@efajardo-nv efajardo-nv deleted the pytorch2-upgrade branch July 5, 2023 20:36
rapids-bot bot pushed a commit that referenced this pull request Jul 6, 2023
- Remove `PIP_FIND_LINKS` for torch in `setup.py`. This was missed by PR #1015 which only removed it in `docker/conda/environments/requirements.txt`.

Authors:
  - Eli Fajardo (https://github.com/efajardo-nv)

Approvers:
  - Michael Demoret (https://github.com/mdemoret-nv)

URL: #1019
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
improvement Improvement to existing functionality non-breaking Non-breaking change
Projects
Archived in project
Development

Successfully merging this pull request may close these issues.

[FEA]: Upgrade to PyTorch 2+
3 participants