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.

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

Fix keras tests for GPU based docker images #413

Open
TRNWWZ opened this issue May 10, 2024 · 0 comments
Open

Fix keras tests for GPU based docker images #413

TRNWWZ opened this issue May 10, 2024 · 0 comments

Comments

@TRNWWZ
Copy link
Contributor

TRNWWZ commented May 10, 2024 •
edited
Loading

Issue: keras tests are failing when running it for GPU .

More background:
We hard code keras to a legacy commit: https://github.com/aws/sagemaker-distribution/blob/main/test/test_artifacts/v1/keras.test.Dockerfile#L19 due to TensorFlow version conflict issue, but with upgrading to new minor version.

After removing this line and rerun the test, test failed again complaining missing dependency like jax.

Acceptance Criteria: keras unit tests should pass.

Image Tags

1.8.0-gpu
This was referenced May 10, 2024
Merged
Closed
@claytonparnell claytonparnell mentioned this issue Jun 25, 2024
Merged
@TRNWWZ TRNWWZ mentioned this issue Jul 17, 2024
Merged
@claytonparnell claytonparnell mentioned this issue Aug 2, 2024
Merged
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
1 participant
@TRNWWZ