Skip to content

[Universal] - install latest cpu torch #1164

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
Merged
merged 6 commits into from
Aug 19, 2024
Merged

[Universal] - install latest cpu torch #1164

merged 6 commits into from
Aug 19, 2024

Conversation

gauravsaini04
Copy link
Contributor

Devcontainer Name

  • Universal

Description

  • This PR attempts to install pytorch 2.4.0 i.e. the latest version with cpu only acceleration limiting it to not use nvidia-cuda and triton packages, thereby reducing the size of the image considerably.

Changelog

  • Added pytorch 2.4.0 along with necessary packages for cpu accelerated latest version of torch package

Checklist

  • Checked that applied changes work as expected

@gauravsaini04 gauravsaini04 marked this pull request as ready for review August 16, 2024 06:24
@gauravsaini04 gauravsaini04 requested a review from a team as a code owner August 16, 2024 06:24
@gauravsaini04 gauravsaini04 changed the title [Universal] - install latest cpu pytorch [Universal] - install latest cpu torch Aug 16, 2024
@samruddhikhandale samruddhikhandale merged commit 2f75804 into devcontainers:main Aug 19, 2024
2 checks passed
@gauravsaini04 gauravsaini04 deleted the universal_install-latest-cpu-pytorch branch August 19, 2024 22:29
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@samruddhikhandale samruddhikhandale samruddhikhandale approved these changes

Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
2 participants
@gauravsaini04 @samruddhikhandale