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

Reduced-size CUDA-Q BYOC image based on pre-built Braket base jobs image #684

Merged
merged 9 commits into from
Mar 10, 2025

Conversation

sterseba
Copy link
Contributor

Issue #, if available: #678

Description of changes:
Replace the Dockerfile for the CUDA-Q BYOC image currently based on the pre-built Braket PyTorch image with a version based on the pre-built Braket base jobs image. This reduces the size of the image on ECR from 13 GB to 4 GB and the time span to pull the image and start the job from more than 6 minutes to about 90 seconds.

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.

@sterseba sterseba requested a review from a team as a code owner February 19, 2025 13:53
@rmshaffer rmshaffer linked an issue Feb 19, 2025 that may be closed by this pull request
@sterseba
Copy link
Contributor Author

Investigating an issue with the example for distributed statevector simulation in example https://github.com/amazon-braket/amazon-braket-examples/blob/main/examples/nvidia_cuda_q/2_parallel_simulations.ipynb.

Copy link

Check out this pull request on  ReviewNB

See visual diffs & provide feedback on Jupyter Notebooks.


Powered by ReviewNB

yitchen-tim
yitchen-tim previously approved these changes Mar 6, 2025
yitchen-tim
yitchen-tim previously approved these changes Mar 6, 2025
rmshaffer
rmshaffer previously approved these changes Mar 6, 2025
@sterseba sterseba dismissed stale reviews from rmshaffer and yitchen-tim via 021ca66 March 7, 2025 04:26
@rmshaffer rmshaffer merged commit 6774846 into amazon-braket:main Mar 10, 2025
9 checks passed
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.

Reduced-size CUDA-Q BYOC image
5 participants