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

TF 2.16 with Torch 2.4.0 #1415

Merged
merged 21 commits into from
Aug 16, 2024
Merged

TF 2.16 with Torch 2.4.0 #1415

merged 21 commits into from
Aug 16, 2024

Conversation

calderjo
Copy link
Contributor

@calderjo calderjo commented Jul 29, 2024

Upgraded our base cpu and gpu base image

-updated torch eco.
-removed patch and pins placed for tf 2.15
-removed torch text, it is no longer maintained and incompatible with torch 2.4.0
-geopanda depreciated it's datasets methods, test needed to be updated

@calderjo calderjo changed the base branch from main to rm-horovod July 29, 2024 22:40
@calderjo calderjo changed the base branch from rm-horovod to main July 29, 2024 22:40
@calderjo calderjo changed the title Tf216 TF 2.16 with Torch 2.4.0 Aug 16, 2024
@calderjo calderjo requested a review from djherbis August 16, 2024 16:07
Copy link
Contributor

@djherbis djherbis left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Awesome work! Feel free to merge and add the other comments in a follow-up, I'd rather get this ready for release :)

Dockerfile.tmpl Show resolved Hide resolved
packages/torch.Dockerfile Show resolved Hide resolved
@calderjo calderjo merged commit 9587f69 into main Aug 16, 2024
4 checks passed
@calderjo calderjo deleted the tf216 branch August 16, 2024 16:18
@calderjo calderjo mentioned this pull request Aug 23, 2024
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.

2 participants