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

feat: upgrade tensorflow to 2.13 #103

Merged
merged 4 commits into from
Sep 22, 2023
Merged

feat: upgrade tensorflow to 2.13 #103

merged 4 commits into from
Sep 22, 2023

Conversation

AbeCoull
Copy link
Contributor

@AbeCoull AbeCoull commented Sep 21, 2023

Issue #, if available:

Description of changes:
Upgrade the TF version as well as the BDK and Pennylane.

Testing done:
Tested using the newly made image.

python3 -m  pytest test/braket_tests/tensorflow --role service-role/AmazonBraketJobsExecutionRole --tag 2.13.0-gpu-py310-cu118-ubuntu20.04-2023-09-22-11-54-56 --repository braket-base-jobs --region us-west-2
============================================================================================================================ test session starts ============================================================================================================================
platform darwin -- Python 3.11.4, pytest-7.2.2, pluggy-1.3.0
rootdir: /Volumes/Unix/workplace/Braket/braket/amazon-braket-containers
plugins: anyio-3.5.0, xdist-3.3.1
collected 1 item

test/braket_tests/tensorflow/test_jobs_qaoa.py .                                                                                                                                                                                                                      [100%]

Merge Checklist

Put an x in the boxes that apply. You can also fill these out after creating the PR. If you're unsure about any of them, don't hesitate to ask. We're here to help! This is simply a reminder of what we are going to look for before merging your pull request.

General

Tests

  • I have added tests that prove my fix is effective or that my feature works (if appropriate)
  • I have checked that my tests are not configured for a specific region or account (if appropriate)

By submitting this pull request, I confirm that my contribution is made under the terms of the Apache 2.0 license.

Comment on lines +39 to +40
PennyLane==0.32.0 \
PennyLane-Lightning==0.32.0 \
Copy link
Contributor

Choose a reason for hiding this comment

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

Should we also upgrade PL in base and pytorch to match?

Copy link
Contributor Author

Choose a reason for hiding this comment

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

As this is an upgrade to the image we pull from, I am scoping this change to only include Tensorflow. There will be a follow up PR to raise those versions as well.

@AbeCoull AbeCoull merged commit b423911 into main Sep 22, 2023
6 checks passed
@AbeCoull AbeCoull deleted the tf_213 branch September 22, 2023 23:13
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