Make libtpu install more explict #7737
Conversation
|
@will-cromar can you take a look at this pr? |
| @@ -34,6 +34,7 @@ To install PyTorch/XLA nightly build in a new TPU VM: | |||
| ``` | |||
| pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cpu | |||
There was a problem hiding this comment.
Regarding #7697 (comment), can you also add an optional command at the end to install compatible torchaudio, torchtext, and torchvision from the PyTorch nightly index?
There was a problem hiding this comment.
we rarely ever need torchaudio or torchtext. I think we put torchvision here because I use resnet in my examples.
There was a problem hiding this comment.
I'll follow up on this since it's tangential to this PR's topic. IMO it's still useful to include the other official ecosystem packages. It's very easy to break nightly torch_xla by installing a stable torchtext that pulls in an incompatible stable torch. We've made this mistake numerous times in our own test cases.
There was a problem hiding this comment.
If we want to install torchaudio, torchtext, etc. from the PyTorch nightly index, the command from PyTorch index page https://screenshot.googleplex.com/8xKE2QXabRsqq9D is pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu. They should be compatible already.
|
@will-cromar can you take another look? |
fix #7731
The libtpu install can be more explicit