build: pin torchvision to latest nightly #1584
Merged
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.
We currently pin the
torch
package to the latest nightly version, butsince
torchvision
depends on thetorch
package, the pip resolverthen has to run through an extensive list of
torchvision
packages thatcan be installed with the pinned
torch
package. This search fails inthe RollPyTorch action, causing pip to settle on an old version of
torchvision
that does not work with our tests. In reality, we areonly interested in a specific version of the
torchvision
package.To make the dependency explicit and to prevent test failures because of
incorrect package installations, this patch makes two key changes:
torchvision
is now pinned to the latest nightly release inpytorch-requirements.txt along with the version of
torch
that isnecessary to install the requested
torchvision
packageThe RollPyTorch action now looks for the latest
torchvision
packageinstead of the latest
torch
package before writing the versionnumbers for pinning in pytorch-requirements.txt