Skip to content

Comments

Add torchvision to requirements-dev.txt#1642

Merged
jeffra merged 2 commits intodeepspeedai:masterfrom
aphedges:require-torchvision
Dec 16, 2021
Merged

Add torchvision to requirements-dev.txt#1642
jeffra merged 2 commits intodeepspeedai:masterfrom
aphedges:require-torchvision

Conversation

@aphedges
Copy link
Contributor

torchvision was reasonably removed from requirements.txt in #1178 because it is not used by the installed version of DeepSpeed. However, it is directly imported in tests/, so it should still be in the dev requirements.

I discovered this when running pip install -U --upgrade-strategy eager -e '.[dev]' to update my development environment. I got a pip warning because torchvision was not updated while torch was. This PR should allow dev installation and updates to now work properly.

Copy link
Collaborator

@jeffra jeffra left a comment

Choose a reason for hiding this comment

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

Good catch, thank you for the contribution.

@jeffra jeffra enabled auto-merge (squash) December 16, 2021 19:31
@jeffra jeffra disabled auto-merge December 16, 2021 19:31
@jeffra jeffra merged commit 8bbf081 into deepspeedai:master Dec 16, 2021
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