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

Move TORCH_VERSION ARG below the FROM #1161

Merged
merged 3 commits into from
May 17, 2022
Merged

Conversation

rosbo
Copy link
Contributor

@rosbo rosbo commented May 17, 2022

This is necessary to make it available with that build context.

The other ARGs are above because they are used in the FROM statement themselves but not the build context of the last image.

I also updated the URL for installing torch_xla. The version of Torch should not include the minor version.

This is necessary to make it available with that build context.

The other ARGs are above because they are used in the FROM statement themselves but not the build context of the last image.
@rosbo rosbo requested a review from djherbis May 17, 2022 18:35

FROM gcr.io/cloud-tpu-v2-images/libtpu:${LIBTPU_IMAGE_TAG} as libtpu
FROM gcr.io/kaggle-images/python-tpu-tensorflow-whl:python-${BASE_IMAGE_TAG}-${TENSORFLOW_VERSION} AS tensorflow_whl
FROM gcr.io/kaggle-images/python:${BASE_IMAGE_TAG}

ARG TORCH_VERSION
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 add a comment explaining this so no one moves it up in the future?

Copy link
Contributor Author

Choose a reason for hiding this comment

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

Good point. Done.

@rosbo
Copy link
Contributor Author

rosbo commented May 17, 2022

I tested running the tpu/build script locally to ensure this now works.

@rosbo rosbo merged commit b03eace into main May 17, 2022
@rosbo rosbo deleted the fix-tensorflow-tpuvm-torch branch May 17, 2022 19:02
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.

None yet

2 participants