Skip to content

chore: restrict torch version to below 2.4 in dependencies (fix the libcudnn dependencies common issue) #1133

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

Open
wants to merge 3 commits into
base: main
Choose a base branch
from

Conversation

jim60105
Copy link
Contributor

@jim60105 jim60105 commented May 4, 2025

A recent commit cd59f21 downgraded ctranslate2 to version 4.4.0, meaning we will remain on cudnn8.

torch depends on libcudnn9 from version 2.4.0 onward.
If we restrict torch<2.4.0, there is no need to manually install libcudnn8 and also save about 1GB disk space.

This PR:

  • Update torch dependency to be below version 2.4.0 instead of at least 2.5.1
  • Change torchaudio dependency to have no minimum version specified
  • restrict onnxruntime to version 1.19 for python 3.9 compatibility

Related #1087, #1158

@jim60105
Copy link
Contributor Author

jim60105 commented Jun 5, 2025

@Barabazs This PR addresses the issue you mentioned in commit 6fe0a87

@jim60105 jim60105 changed the title chore: restrict torch version to below 2.4 in dependencies chore: restrict torch version to below 2.4 in dependencies (fix the libcudnn dependencies common issue) Jun 13, 2025
jim60105 added 3 commits June 14, 2025 00:21
torch depends on libcudnn9 from version 2.4.0 onward.
If we restrict torch<2.4.0, there is no need to manually install libcudnn8 and also save about 1GB disk space.

- Update torch dependency to be below version 2.4.0 instead of at least 2.5.1
- Change torchaudio dependency to have no minimum version specified

Signed-off-by: CHEN, CHUN <[email protected]>
- Restrict the onnxruntime dependency to versions >=1.19 and <1.20.0 to avoid potential compatibility issues.

Signed-off-by: CHEN, CHUN <[email protected]>
…n README"

This reverts commit 6fe0a87.

Revert the commit now that the issue is fixed.

Signed-off-by: CHEN, CHUN <[email protected]>
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.

1 participant