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

Resolve Torch version by determining related dependencies #1948

Open
8W9aG opened this issue Sep 12, 2024 · 0 comments
Open

Resolve Torch version by determining related dependencies #1948

8W9aG opened this issue Sep 12, 2024 · 0 comments
Assignees

Comments

@8W9aG
Copy link
Contributor

8W9aG commented Sep 12, 2024

Currently if we have a cog.yaml that defines the following:

torch
torchvision==0.19.0+cu121

The build fails. The reason being that we always resolve a versionless torch to the maximum we support in our base images (which is currently 2.4.1). The torchvision is pinned with version 0.19.0 which causes version incompatibility due to 2.4.1 not being compatible with torchvision 0.19.0. In our compatibility matrix we do define what torchvisions are compatible with what versions of torch, so when we resolve the torch version we should take this data into account. We also need to do this for torchaudio which we similarly have version info about.

@8W9aG 8W9aG self-assigned this Sep 12, 2024
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

No branches or pull requests

1 participant