Update TPU docs for installation #6794
Merged
Conversation
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
kaushikb11
requested review from
Borda,
awaelchli,
edenlightning and
tchaton
as code owners
Borda
reviewed
Apr 1, 2021
| !curl https://raw.githubusercontent.com/pytorch/xla/master/contrib/scripts/env-setup.py -o pytorch-xla-env-setup.py | ||
| !python pytorch-xla-env-setup.py --version 1.7 --apt-packages libomp5 libopenblas-dev | ||
| !pip install cloud-tpu-client==0.10 https://storage.googleapis.com/tpu-pytorch/wheels/torch_xla-1.8-cp37-cp37m-linux_x86_64.whl |
Collaborator
There was a problem hiding this comment.
well here the XLA version shall be aligned with PT version, right?
Contributor
Author
There was a problem hiding this comment.
The instructions are for Colab & they have updated the runtime to use torch>=1.8.0
Contributor
There was a problem hiding this comment.
does this same command work for kaggle?
Contributor
Author
There was a problem hiding this comment.
@lezwon Just checked. It's 1.7 for Kaggle notebooks.
We could add {torch.__version__} to the URL as a placeholder.
Codecov Report
@@ Coverage Diff @@
## master #6794 +/- ##
=======================================
- Coverage 92% 87% -5%
=======================================
Files 192 192
Lines 12174 12248 +74
=======================================
- Hits 11147 10618 -529
- Misses 1027 1630 +603 |
Borda
approved these changes
Apr 1, 2021
awaelchli
approved these changes
Apr 2, 2021
akihironitta
approved these changes
Apr 3, 2021
Closed
kaushikb11
added a commit
to kaushikb11/pytorch-lightning
that referenced
this pull request
Apr 6, 2021
lexierule
pushed a commit
that referenced
this pull request
Apr 7, 2021
* update readme by v1.2.x (#6728) * [bugfix] Add support for omegaconf and tpu (#6741) * fix_hydra * update changelog Co-authored-by: Your Name <[email protected]> * [docs] Update Bolts link (#6743) * Update Bolts link * Update Bolts link * formt Co-authored-by: Jirka Borovec <[email protected]> * Update logic for checking TPUs availability (#6767) * Update logic for checking TPUs availability * fix flake8 * add fix * resolve bug (#6781) * Fix validation progress counter with check_val_every_n_epoch > 1 (#5952) Co-authored-by: rohitgr7 <[email protected]> Co-authored-by: Carlos Mocholí <[email protected]> * Remove extinct parameters from lightning_module.rst (#6801) Fixes #6800 * Update TPU docs for installation (#6794) * fix boolean check on iterable dataset when len not defined (#6828) * fix iterable dataset len check * update predict and validate * add validate to test * add changelog * add predict * Sanitize `None` params during pruning (#6836) * sanitize none params during pruning * amend * Fix `unfreeze_and_add_param_group` expects `modules` rather than `module` (#6822) * Enforce an epoch scheduler interval when using SWA (#6588) Co-authored-by: Carlos Mocholi <[email protected]> * Fix DPP + SyncBN (#6838) * Fix DPP + SyncBN Ensure that model is already on correct GPU before applying SyncBN conversion * Fix order of SyncBN for ddp_spawn * [Fix] TPU Training Type Plugin (#6816) * Fix support for symlink save_dir in TensorBoardLogger (#6730) * Add test for symlink support and initial fix * Respond to comment and add docstring * Update CHANGELOG.md * Simplify * Update pytorch_lightning/utilities/cloud_io.py Co-authored-by: Carlos Mocholí <[email protected]> * Make `LightningLocalFileSystem` protected Co-authored-by: Carlos Mocholí <[email protected]> * Fixed missing arguments in `lr_find` call (#6784) There seem to be 3 arguments missing in the `lr_find` call in the tunining.py file. * Update Changelog & version * Fix TPU tests for checkpoint Skip advanced profiler for torch > 1.8 Skip pytorch profiler for torch > 1.8 Fix save checkpoint logic for TPUs Co-authored-by: Jirka Borovec <[email protected]> Co-authored-by: thomas chaton <[email protected]> Co-authored-by: Your Name <[email protected]> Co-authored-by: Akihiro Nitta <[email protected]> Co-authored-by: Yuan-Hang Zhang <[email protected]> Co-authored-by: rohitgr7 <[email protected]> Co-authored-by: Carlos Mocholí <[email protected]> Co-authored-by: Elizaveta Logacheva <[email protected]> Co-authored-by: Adrian Wälchli <[email protected]> Co-authored-by: Karthik Prasad <[email protected]> Co-authored-by: Sadiq Jaffer <[email protected]> Co-authored-by: Michael Baumgartner <[email protected]> Co-authored-by: Eugene Khvedchenya <[email protected]> Co-authored-by: Ethan Harris <[email protected]> Co-authored-by: Tharindu Hasthika <[email protected]>
lexierule
pushed a commit
that referenced
this pull request
Apr 7, 2021
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
5 participants
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
What does this PR do?
Ref: pytorch/xla#2813
Before submitting
PR review
Anyone in the community is free to review the PR once the tests have passed.
Before you start reviewing make sure you have read Review guidelines. In short, see the following bullet-list:
Did you have fun?
Make sure you had fun coding 🙃